Yolo_v1-pytorch
项目介绍
整体框架作者：杨航（759979738@qq.com，西南民族大学 电子信息学院 人工智能实验室 608）
代码优化作者：徐钒鑫（xufanxin86@gmail.com，西南民族大学 电子信息学院 人工智能实验室 608）
文件分布
data_convert.py
负责数据集的预处理
train.py
负责数据的训练
inference.py
用训练好的模型推理
三个文件 均 单独运行
数据集
VOC2012
环境
pytorch 1.7.1
整体框架作者：杨航（759979738@qq.com，西南民族大学 电子信息学院 人工智能实验室 608）
代码优化作者：徐钒鑫（xufanxin86@gmail.com，西南民族大学 电子信息学院 人工智能实验室 608）
data_convert.py
负责数据集的预处理
train.py
负责数据的训练
inference.py
用训练好的模型推理
三个文件 均 单独运行
VOC2012
pytorch 1.7.1