【文件属性】:
文件名称:MATLAB生成cuda代码使用-YOLO:YOLO
文件大小:8KB
文件格式:ZIP
更新时间:2021-05-27 08:29:32
系统开源
MATLAB生成cuda代码使用
YOLO
安装
torch-1.2.0-cp36-cp36m-win_amd64.whl
torchvision-0.4.0-cp36-cp36m-win_amd64.whl
CUDA
10.1
该文件夹存放使用pytorch实现的代码版本
model.py:
是模型文件
train.py:
是调用模型训练的文件
predict.py:
是调用模型进行预测的文件
class_indices.json:
是训练数据集对应的标签文件
(1)在data_set文件夹下创建新文件夹"flower_data"
(2)打开flower_link.txt文档,复制网址到浏览器会自动进行下载花分类数据集
(3)解压数据集到flower_data文件夹下
(4)执行"split_data.py"脚本自动将数据集划分成训练集train和验证集val
(不要重复使用该脚本,否则训练集和验证集会混在一起,flower_data文件夹结构如下)
|——
flower_data
|————
flower_photos(解压的数据集文件夹,3670个样本)
|————
train(
【文件预览】:
YOLO-master
----class_indices.json(145B)
----train.py(5KB)
----飞飞裁剪代码.txt(1KB)
----fog25ReadTxt.m(591B)
----cutImage_yanzhen.m(755B)
----FEIFEIseach.py(2KB)
----model.py(2KB)
----predict.py(2KB)
----README.md(1KB)