首页 技术 正文
技术 2022年11月21日
0 收藏 752 点赞 4,170 浏览 661 个字

数据集做好后,训练程序为/examples/ssd/ssd_pascal.py,运行之前,我们需要修改相关路径代码,主要是训练路径的修改和关于自己数据集参数的一些修改.

cd /examples/ssd
先复制一份sd_pascal.py, 再打开,把里面的路径修改为自己的.具体有如下:
train_data路径;
test_data路径;
save_dir、snapshot_dir、job_dir、output_result_dir路径;没有的话就仿照voc0712的文件夹进行建立.
name_size_file、label_map_file路径;

name_size_file

label_map_file
num_classes 修改为1 + 类别数
num_test_image:测试集图片数目(即/home/zhuyating/caffessd/caffe/data/zyt目录下的test.txt文件夹内的图片信息个数)

batch_size根据自己电脑性能修改,因为本人的环境只有一个gpu(华硕1080GPU),所以将其改为1,以加快训练速度。

另外, 如果你只有一个GPU, 需要修改285行:(不同人的行数不一定都在285行) 
gpus=”0,1,2,3” ===> 改为”0”。
否则,训练的时候会出错。

另外,自己的电脑没有GPU的话,最好不要训练,一训练可能就死机。

修改完后运行

python ./examples/ssd/ssd_pascal.py

训练完, 修改ssd_detector.py中模型路径, 任意找一张图片识别,看看效果怎么样。

相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:9,077
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,552
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,400
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,176
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:7,813
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:4,894