nqt228 / YoloZ

YoloZ

Training

$ python train.py --data coco.yaml --cfg dense.yaml --weights '' --batch-size 64

Citation

@article{benjumeayoloz,
  title={YOLO-Z: Improving small object detection in YOLOv5 for autonomous vehicles,
  author={Aduen Benjumea, Izzeddin Teeti, Fabio Cuzzolin, Andrew Bradley},
  journal={arXiv preprint arXiv:2112.11798},
  year={2021}
}

REFERENCE

[1] https://arxiv.org/pdf/2112.11798.pdf

[2] https://github.com/ultralytics/yolov5