Selaa lähdekoodia

fix arc dataset

zhaoyinghan 4 viikkoa sitten
vanhempi
commit
a456dade20
1 muutettua tiedostoa jossa 1 lisäystä ja 0 poistoa
  1. 1 0
      models/line_detect/line_dataset.py

+ 1 - 0
models/line_detect/line_dataset.py

@@ -137,6 +137,7 @@ class LineDataset(BaseDataset):
 
 
 
+
         target["boxes"] = boxes
         target["labels"] = labels
         # target["boxes"], lines,target["points"], target["labels"] = get_boxes_lines(objs,shape)