Zalo AI challenge 2021 5K-Compliance
Prepare data: Dữ liệu của ban tổ chức cung cấp nằm trong folder dataset/train. Kết quả sau khi chia chính xác như các file txt
- Clone yolov5 và mô hình yolov5x6 và thay thế file detect.py
git clone https://github.com/ultralytics/yolov5
scp detect.py yolov5
cd yolov5
wget https://github.com/ultralytics/yolov5/releases/download/v6.0/yolov5x6.pt
python detect.py --weights yolov5x6.pt --source /dataset/train/images --img 1536 --conf-thres 0.1 --iou-thres 0.7 --nosave --classes 0 --augment
cd ..
python3 prepare_mask_dataset.py
python3 prepare_distance_dataset.py
train mask model
bash script_train_mask/runtrain1.sh
bash script_train_mask/runtrain2.sh
bash script_train_mask/runtrain3.sh
bash script_train_mask/runtrain4.sh
bash script_train_mask/runtrain5.sh
train distance model
bash script_train_distance/runtrain1.sh
bash script_train_distance/runtrain2.sh
bash script_train_distance/runtrain3.sh
bash script_train_distance/runtrain4.sh
bash script_train_distance/runtrain5.sh
inference
1 May 29, 2022
904 Jan 02, 2023
7.9k Jan 02, 2023
94 Nov 14, 2022
226 Jan 02, 2023
2 Nov 23, 2021
255 Dec 28, 2022
3 Mar 14, 2022
3.4k Jan 08, 2023
2 Dec 27, 2021
1 Oct 23, 2021
2 Jan 20, 2022
24 Oct 05, 2022
545 Jan 07, 2023
7 Dec 02, 2021
90 Jan 01, 2023
199 Dec 27, 2022
1 Dec 20, 2021
14 Sep 03, 2022
37 Nov 21, 2022