henry000 commited on
Commit
a7a9479
·
1 Parent(s): 631540a

🦺 [Add] Mock test set for testing dataloader

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. yolo/config/dataset/mock.yaml +17 -0
.gitignore CHANGED
@@ -114,6 +114,7 @@ dmypy.json
114
  runs
115
  /data
116
  /datasets
 
117
 
118
  # Datasets and model checkpoints
119
  *.pth
 
114
  runs
115
  /data
116
  /datasets
117
+ */data
118
 
119
  # Datasets and model checkpoints
120
  *.pth
yolo/config/dataset/mock.yaml ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ path: tests/data
2
+ train: train
3
+ validation: val
4
+
5
+ auto_download:
6
+ images:
7
+ base_url: https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/
8
+ train:
9
+ file_name: mock_train
10
+ file_num: 5
11
+ val:
12
+ file_name: mock_val
13
+ file_num: 5
14
+ annotations:
15
+ base_url: https://github.com/WongKinYiu/yolov9mit/releases/download/v1.0-alpha/
16
+ annotations:
17
+ file_name: mock_annotations