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
603 Jan 03, 2023
304 Jan 03, 2023
5 Apr 30, 2022
78 Dec 19, 2022
163 Oct 14, 2022
6 Nov 29, 2022
13 Dec 08, 2022
2.6k Jan 06, 2023
159 Dec 09, 2022
56 Dec 22, 2022
41 Sep 10, 2022
44 Nov 12, 2022
58 Nov 07, 2022
4 Oct 02, 2022
3 Feb 16, 2022
7 Jan 06, 2022
153 Dec 28, 2022
1 Nov 30, 2021
716 Dec 26, 2022
2 Oct 25, 2021