zoukankan      html  css  js  c++  java
  • 使用SSD检测框架训练自己的数据

    数据集做好后,训练程序为/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中模型路径, 任意找一张图片识别,看看效果怎么样。

  • 相关阅读:
    构建之法 读书笔记二
    聚集索引,非聚集索引,覆盖索引
    最佳左前缀法则
    悲观锁和乐观锁
    JVM (三)- GC 垃圾回收器
    JVM 内存模型
    Java内存模型(JMM) 和 JVM 内存模型区别
    byType 和 byName 的区别
    Spring注入方式
    Java高性能编程-java基础-1.1.5线程通信
  • 原文地址:https://www.cnblogs.com/wmlj/p/8681650.html
Copyright © 2011-2022 走看看