✅ [Pass] test and fix typo in v7 yaml
Browse files
tests/test_tools/test_data_loader.py
CHANGED
|
@@ -35,7 +35,7 @@ def test_training_data_loader_correctness(train_dataloader: YoloDataLoader):
|
|
| 35 |
Path("tests/data/images/train/000000050725.jpg"),
|
| 36 |
Path("tests/data/images/train/000000167848.jpg"),
|
| 37 |
]
|
| 38 |
-
assert image_paths == expected_paths
|
| 39 |
|
| 40 |
|
| 41 |
def test_validation_data_loader_correctness(validation_dataloader: YoloDataLoader):
|
|
@@ -50,7 +50,7 @@ def test_validation_data_loader_correctness(validation_dataloader: YoloDataLoade
|
|
| 50 |
Path("tests/data/images/val/000000323571.jpg"),
|
| 51 |
Path("tests/data/images/val/000000570456.jpg"),
|
| 52 |
]
|
| 53 |
-
assert image_paths == expected_paths
|
| 54 |
|
| 55 |
|
| 56 |
def test_file_stream_data_loader_frame(file_stream_data_loader: StreamDataLoader):
|
|
|
|
| 35 |
Path("tests/data/images/train/000000050725.jpg"),
|
| 36 |
Path("tests/data/images/train/000000167848.jpg"),
|
| 37 |
]
|
| 38 |
+
assert list(image_paths) == list(expected_paths)
|
| 39 |
|
| 40 |
|
| 41 |
def test_validation_data_loader_correctness(validation_dataloader: YoloDataLoader):
|
|
|
|
| 50 |
Path("tests/data/images/val/000000323571.jpg"),
|
| 51 |
Path("tests/data/images/val/000000570456.jpg"),
|
| 52 |
]
|
| 53 |
+
assert list(image_paths) == list(expected_paths)
|
| 54 |
|
| 55 |
|
| 56 |
def test_file_stream_data_loader_frame(file_stream_data_loader: StreamDataLoader):
|
yolo/config/model/v7.yaml
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
name: v7
|
| 2 |
|
| 3 |
anchor:
|
| 4 |
anchor:
|
|
|
|
| 1 |
+
name: v7
|
| 2 |
|
| 3 |
anchor:
|
| 4 |
anchor:
|