Skip to content

Commit b9d076f

Browse files
committed
[TEST] Yolo nested tests
1 parent 3d85c14 commit b9d076f

File tree

4 files changed

+8
-1
lines changed

4 files changed

+8
-1
lines changed

.vscode/settings.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,6 @@
1111
"editor.defaultFormatter": "charliermarsh.ruff",
1212
},
1313
"notebook.formatOnSave.enabled": false,
14+
"python.testing.pytestEnabled": true,
15+
"python.testing.pytestArgs": ["tests"],
1416
}

tests/constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
coco2_path = gts_path / "coco_format_v2/instances_v2.json"
1414
coco_gts_path = data_path / "coco_eval/ground_truths.json"
1515
yolo_path = gts_path / "yolo_format/obj_train_data"
16+
yolo_nested_path = gts_path / "yolo_nested_format/obj_train_data"
1617
yolo_seg_path = gts_path / "yolo_seg_format/obj_train_data"
1718
cvat_path = gts_path / "cvat_format/annotations.xml"
1819
imagenet_path = gts_path / "imagenet_format/Annotations/"

tests/globox_test_data

Submodule globox_test_data updated 103 files

tests/test_parsing.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,9 @@ def tests_parsing():
1212
yolo_set = AnnotationSet.from_yolo_darknet(
1313
C.yolo_path, image_folder=C.image_folder
1414
).map_labels(C.id_to_label)
15+
yolo_nested_set = AnnotationSet.from_yolo_darknet(
16+
C.yolo_nested_path, image_folder=C.image_folder, recursive=True
17+
).map_labels(C.id_to_label)
1518
yolo_seg_set = AnnotationSet.from_yolo_seg(
1619
folder=C.yolo_seg_path, image_folder=C.image_folder
1720
).map_labels(C.id_to_label)
@@ -49,6 +52,7 @@ def tests_parsing():
4952
coco2_set,
5053
coco3_set,
5154
yolo_set,
55+
yolo_nested_set,
5256
yolo_seg_set,
5357
cvat_set,
5458
imagenet_set,

0 commit comments

Comments
 (0)