Fix yolo test

hovnatan · hovnatan · commit 9c544e837c6e · 2021-01-15T12:56:51.000+04:00
diff --git a/superannotate/input_converters/converters/yolo_converters/yolo_to_sa_vector.py b/superannotate/input_converters/converters/yolo_converters/yolo_to_sa_vector.py
@@ -43,7 +43,7 @@ def yolo_object_detection_to_sa_vector(data_path, output_dir):
             )
             continue
 
-        if Path(files_list[0]).suffix == 'txt':
+        if Path(files_list[0]).suffix == '.txt':
             file_name = files_list[1]
         else:
             file_name = files_list[0]

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@ def yolo_object_detection_to_sa_vector(data_path, output_dir):`
`43`	`43`	`)`
`44`	`44`	`continue`
`45`	`45`
`46`		`- if Path(files_list[0]).suffix == 'txt':`
	`46`	`+ if Path(files_list[0]).suffix == '.txt':`
`47`	`47`	`file_name = files_list[1]`
`48`	`48`	`else:`
`49`	`49`	`file_name = files_list[0]`