diff --git a/config/datasets/sibur_people_n_head_detection.yaml b/config/datasets/sibur_people_n_head_detection.yaml index 4725bd70..9634d659 100644 --- 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 index da934729..98b09322 100755 --- 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 index 61a91f93..7bfb330a 100644 --- 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 \ No newline at end of file +python infer.py experiments=KA_170122_43fdja_yolov5_sibur +ckpt_path=https://api.blackhole.ai.innopolis.university/pretrained/best.pt \ No newline at end of file