Skip to content

Commit 4a18390

Browse files
authored
Update README.md
1 parent 2e86c64 commit 4a18390

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

README.md

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,18 @@
11
# PyTorch_YOLOv1
2-
本项目代码是服务于本人的YOLO入门教程的知乎专栏:《YOLO-从零开始入门目标检测》,网页链接如下
2+
这个YOLOv1项目是配合我在知乎专栏上连载的《YOLO入门教程》而创建的
33

4-
https://www.zhihu.com/column/c_1364967262269693952
4+
https://zhuanlan.zhihu.com/c_1364967262269693952
5+
6+
感兴趣的小伙伴可以配合着上面的专栏来一起学习,入门目标检测。
7+
8+
另外,这个项目在小batch size 的情况,如batch size=8,可能会出现nan的问题,经过其他伙伴的调试,
9+
在batch size=8时,可以把学习率lr跳到2e-4,兴许就可以稳定炼丹啦! 我自己训练的时候,batch size
10+
设置为16或32,比较大,所以训练稳定。
11+
12+
当然,这里也诚挚推荐我的另一个YOLO项目,训练更加稳定,性能更好呦
13+
14+
https://github.com/yjh0410/PyTorch_YOLO-Family
515

6-
欢迎学习和交流!
716

817
## 网络结构
918

0 commit comments

Comments
 (0)