-
new-YOLOv1_PyTorch下载
资源介绍
new-YOLOv1_PyTorch
在这个项目中，您可以享受：
yolov1的新版本
网络
这是PyTorch构建的YOLOv1的新版本：
骨干网：resnet18
负责人：SPP，SAM
火车
批量大小：32
基础LR：1E-3
最多纪元：160
LRstep：60、90
优化器：SGD
在告诉您如何使用该项目之前，我必须说一件事，关于起源yolo-v2与我的yolo-v2之间的区别：
对于数据扩充，我从复制了扩充代码，这是一个复制SSD的极好的项目。 如果有人对SSD感兴趣，只需克隆即可学习！（别忘了给它加注星标！）
所以我不会自己写数据扩充。 我有点懒~~
我的损失函数和tools.py创建者都在tools.py ，您可以尝试更改任何参数以改进模型。
实验
环境：
Python3.6，opencv-python，PyTorch1.1.0，CUDA10.0，