Skip to content

itouchzh/YOLO_ED

Folders and files

NameName
Last commit message
Last commit date

Latest commit

9bea16b · Feb 17, 2024

History

7 Commits
Jun 6, 2023
Feb 14, 2024
Oct 31, 2023
Feb 17, 2024
Jun 6, 2023
Feb 14, 2024
Oct 31, 2023
Jun 6, 2023
Jun 6, 2023
Jun 6, 2023
Jun 6, 2023
Oct 31, 2023
Jun 17, 2023
Jun 6, 2023
Feb 14, 2024
Feb 14, 2024
Jun 6, 2023

Repository files navigation

EDN-YOLO✈️

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 and Detect:

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

Detection results

YOLOv5s EDN-YOLO YOLOv5s EDN-YOLO

About

Traffic sign detection using yolo

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published