整体框架作者：杨航（[email protected]，西南民族大学 电子信息学院 人工智能实验室 608）
代码优化作者：徐钒鑫（[email protected]，西南民族大学 电子信息学院 人工智能实验室 608）
data_convert.py
负责数据集的预处理
train.py
负责数据的训练
inference.py
用训练好的模型推理
三个文件 均 单独运行
VOC-2012
pytorch 1.7.1
python 3.7.9
-
Notifications
You must be signed in to change notification settings - Fork 0
jaheel/YOLO_v1
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
implements YOLOv1 with pytorch (overall framework : Yang Hang, detail optimization : Alex)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published