Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
IT从业者张某某
yolo_v3
提交
9efceef6
yolo_v3
项目概览
IT从业者张某某
/
yolo_v3
与 Fork 源项目一致
Fork自
DataBall / yolo_v3
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
yolo_v3
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
9efceef6
编写于
3月 22, 2021
作者:
DataBall
🚴🏻
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
添加了 train.py 配置文件的说明
上级
e545451b
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
25 addition
and
0 deletion
+25
-0
README.md
README.md
+25
-0
未找到文件。
README.md
浏览文件 @
9efceef6
...
...
@@ -126,5 +126,30 @@ h = h*dh
### 模型训练
*
根目录下运行命令: python train.py (注意脚本内相关参数配置 )
*
注意:加载训练的配置参数的关键代码段如下,cfg/transport.data为训练coco交通工具数据集的配置参数文件。
```
if __name__ == '__main__':
train(data_cfg = "cfg/transport.data")
```
*
cfg文件夹下的 "
*
.data"文件包含了重要的配置信息,以"cfg/transport.data"为例如下:
```
cfg_model=yolo # 模型选择:yolo / yolo tiny
classes=8 # 数据集的类别
gpus = 0 # GPU 的选择
num_workers = 8 # 训练时数据迭代(增强)器的进程数
batch_size = 8 # 训练时每一个批次的图片量设定,该例子是8张图片
img_size = 416 # 模型的默认输入图片尺寸是 416*416
multi_scale = True # 是否在训练中采用多尺度增强方式,该例子为使用
epochs = 100 # 训练总周期,完整遍历一次数据集为一个周期
train=./yolo_transport_train/anno/train.txt # 训练集的train.txt路径
valid=./yolo_transport_train/anno/train.txt # 验证集的train.txt路径,目前训练代码中并没有加入验证功能代码
names=./cfg/transport.names # 该数据集对应的具体分类名字,需要与其标签对应
finetune_model = ./weights-yolov3-transport/latest_416.pt # 采用的预训练模型
lr_step = 10,20,30 # 训练时对应的特定epoch时更新学习率
lr0 = 0.001 # 训练时的初始化学习率
```
### 模型推理
*
根目录下运行命令: python predict.py (注意脚本内相关参数配置 )
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录