Merge pull request InnopolisUni#91 from QazyBi/feature/add-sibur-data

updated weights path
QazyBi · Jan 17, 2023 · 50496f4 · 50496f4
2 parents 61b9eed + f47a24c
commit 50496f4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/config/datasets/sibur_people_n_head_detection.yaml b/config/datasets/sibur_people_n_head_detection.yaml
@@ -27,7 +27,7 @@ num_workers: 8
 
 val_size: 0.2
 channels_num: 3
-image_size: 600
+image_size: 1440
 num_classes: 2
 names:
   - people

diff --git a/config/models/yolov5.yaml b/config/models/yolov5.yaml
@@ -2,7 +2,7 @@ name: yolov5
 description: yolov5 by ultralytics
 
 _target_: innofw.core.models.torch.architectures.detection.YOLOv5
-arch: yolov5s  # x
+arch: yolov5x  # s x
 num_classes: 4
 
 

diff --git a/examples/infer_people_n_head_detection.sh b/examples/infer_people_n_head_detection.sh
@@ -1 +1 @@
-python infer.py experiments=KA_170122_43fdja_yolov5_sibur +ckpt_path=logs/train/people_n_head_detection/KA_170122_43fdja_yolov5_sibur/20230117-065904/weights/best.pt
+python infer.py experiments=KA_170122_43fdja_yolov5_sibur +ckpt_path=https://api.blackhole.ai.innopolis.university/pretrained/best.pt
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		python infer.py experiments=KA_170122_43fdja_yolov5_sibur +ckpt_path=logs/train/people_n_head_detection/KA_170122_43fdja_yolov5_sibur/20230117-065904/weights/best.pt
		python infer.py experiments=KA_170122_43fdja_yolov5_sibur +ckpt_path=https://api.blackhole.ai.innopolis.university/pretrained/best.pt