search

BumbleBee-BBStream / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
0 stars 0 forks source link

14_defect_V6.5_JinChuan #23

Open Samuel-wei opened 1 year ago

Samuel-wei commented 1 year ago

September 21, 2023，晋川数据训练--Small detect

Training Version: 14_defect_6.5

Camera-0123, total:

  1. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6.3/train_images(13160)&train_xmls(13160)
  2. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/14_FP_V6.0.4(2510 xml,jpg)
  3. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/14_FP_V6.1.1_N1-N4(1700 xml,jpg)
  4. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/14_FP_V6.3.0_N1(664 xml,jpg)

Validation dataset

Camera-0123, total: 1477

  1. /home/workspace/BatteryDetect/xmldata/small_defect_JinChuan_V5/test_14_20220803_20220802(1900)
  2. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/14_FP_V6.1.1(522_xml,jpg)

Training On HuaWei

$ docker exec -it yolov7_pytorch_devel python3 train.py --epochs 100 --workers 4 --device 0 --batch-size 32 --data data/battery.yaml --img 640 640 --cfg cfg/training/battery-tiny.yaml --weights '/home/workspace/yolo_weights/yolov7-tiny.pt' --name 14_defect_V6.5 --hyp data/hyp.scratch.tiny.yaml

Result

Image

Test-dev

  1. /home/workspace/BatteryDataSet/Data_xml/TestData/NG_14_20220907(448) $ docker exec -it yolov7_pytorch_deve_ssh python3 BatteryDetect.py --weights /home/workspace/BatteryDetect/models/14_defect_V6.5/weights/best.pt --conf 0.3 --img-size 640 --device 0 --source /home/workspace/BatteryDataSet/Data_xml/TestData/NG_14_20220907

Image

  1. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/val_14_FP_V6.1.1_N5 $ docker exec -it yolov7_pytorch_deve_ssh python3 BatteryDetect.py --weights /home/workspace/BatteryDetect/models/14_defect_V6.5/weights/best.pt --conf 0.3 --img-size 640 --device 0 --source /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/val_14_FP_V6.1.1_N5

Image

  1. /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/small_14_ok_V5.1.0_tp $ docker exec -it yolov7_pytorch_deve_ssh python3 BatteryDetect.py --weights /home/workspace/BatteryDetect/models/14_defect_V6.5/weights/best.pt --conf 0.3 --img-size 640 --device 0 --source /home/workspace/BatteryDataSet/Data_xml/small_defect_JinChuan_V6/small_14_ok_V5.1.0_tp

Image

Samuel-wei commented 9 months ago

January 31, 2024

$ docker exec -it yolov7_pytorch_devel python3 train.py --epochs 100 --workers 4 --device 0 --batch-size 8 --data data/battery.yaml --img 640 640 --cfg cfg/training/yolov7_battery.yaml --weights /home/workspace/yolo_weights/yolov7.pt --name 14_defect_V6.5_JinChuan --hyp data/hyp.scratch.custom.yaml

--weights /home/workspace/yolo_weights/yolov7.pt