We proposed EDN-YOLO, a model for multi-scale traffic sign detection, which has been verified on different data sets and achieved good detection results.
Train
python train.py --data TT100K.yaml --epochs 150 --weights '' --cfg yolov5s.yaml --batch-size 32
Val
python val.py --data TT100K.yaml --weights 'xxxx' --cfg yolov5s.yaml --batch-size 1
Inference with detect.py
python detect.py --weights xxx.pt path/ # directory
YOLOv5s | EDN-YOLO | YOLOv5s | EDN-YOLO |
---|---|---|---|