Robustness object detection in adverse weather using yolov5

DAWN dataset rain_storm-203.jpg

Dataset

• BDD100k(Train, valid)
• DAWN(Test)

Model

Using 'Yolov5m.pt' pre-trained weights

How To Run ✨

install

git clone https://github.com/ultralytics/yolov5  # clone
cd yolov5
pip install -r requirements.txt  # install

Train

Example

python train_v4.py --data bdd100k2.yaml --weights yolov5m.pt --img 640 --epochs 50 --batch-size 8 --image_weights

Evaluation

Example

python yolov5/detect.py --weights yolov5/runs/train/exp4/weights/best.pt --source dataset/DAWN/val --img 640 --project yolov5/runs/test

Reference

YOLOV5🚀|GitHub

Updating...