darknet框架下训练YOLO

YOLO训练步骤

配置文件

  1. 更改darknet\cfg\ 下配置文件中classesfilters(# or 255) 最后的feature map每个像素预测num 个框
    filters = v3 = ((num=9) / 3) * (classes + 5); v2 = num * (classes + 5)
  2. 更改cfg\voc.data 键入一些路径
  3. 更改data\voc.names 你的类别名称
  4. data\train_list.txt 生成路径列表文件,用vim 打开后将格式设置为set fileformat = unix
  5. example\detector.c 中修改存储模型的间隔

训练语句

  1. 训练命令
./darknet detector train cfg/voc.data cfg/yolov3-spp.cfg darknet53.conv.74   # 训练命令
  1. 断点开始
./darknet detector train cfg/voc.data cfg/yolov3-spp.cfg backup/yolov3-spp.backup #断点重新开始
  1. 查看摄像头
ls  /dev/video*  查看摄像头
  1. 生成log文件
 | tee train_yolov3.log

生成训练列表train_list

import os

def file_name(file_dir):
	with open('train_list.txt', 'w') as f:
		for a, b, files in os.walk(file_dir):
			
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值