文件名称:new-YOLOv1_PyTorch
文件大小:36KB
文件格式:ZIP
更新时间:2024-05-28 09:14:24
Python
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,
【文件预览】:
new-YOLOv1_PyTorch-master
----models()
--------yolo.py(7KB)
----eval.py(3KB)
----tools.py(4KB)
----data()
--------cocodataset.py(13KB)
--------__init__.py(1KB)
--------voc0712.py(12KB)
--------config.py(142B)
--------scripts()
----test.py(5KB)
----train.py(11KB)
----backbone()
--------resnet.py(7KB)
--------__init__.py(70B)
----weights()
--------README.md(381B)
----README.md(5KB)
----utils()
--------augmentations.py(13KB)
--------__init__.py(65B)
--------cocoapi_evaluator.py(5KB)
--------vocapi_evaluator.py(13KB)
--------modules.py(3KB)
--------com_paras_flops.py(561B)