Update README.md

XingZeng307 · Apr 28, 2021 · 058dbf1 · 058dbf1
1 parent 2489b3a
commit 058dbf1
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -8,3 +8,14 @@ This repo is mainly for replacing PANet with BiFPN in YOLOv5, which you can chec
 python train.py --img 640 --batch 8 --epochs 200 --data CUB.yaml --weights '' --cfg yolov5x.yaml
 
 I didn't use pre-trained weights since the architecture changes a bit. And as for the performance, I will check more dataset and update it.
+
+
+# Reference
+
+https://github.com/ultralytics/yolov5
+
+https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch/tree/15403b5371a64defb2a7c74e162c6e880a7f462c
+
+Mingxing Tan, Ruoming Pang, and Quoc V Le. EfficientDet: Scalable and efficient object detection. In Proceedings
+of the IEEE Conference on Computer Vision and Pattern
+Recognition (CVPR), 2020.