zoukankan      html  css  js  c++  java
  • 使用faster-rcnn.pytorch训练自己数据集(简洁版)

    1、从这里下载FasterRCNN的Pytorch版本代码。按照提示完成相关编译。

    2、因为默认的数据为VOC2007,最直接的方法我们可以按照这里将自己的数据制作成VOC2007的格式,并建立软连接。

    3、修改pascal_voc.py数据的类别,即将self._classes修改为自己的类。注:保留background类,用于训练RPN网络,视作RPN网络的负样本。

    4、运行trainval_net.py即可。

    5、测试时,需要将test_net.py的类别做相应修改。

    注:① 在每次训练前,需要将data下面的cache删掉。

           ② 如果跑demo.py会发现检测不完整,这是因为demo里最多显示10个检测框,修改响应代码即可。
    ---------------------
    ref:https://blog.csdn.net/yutingzhaomeng/article/details/80328422

  • 相关阅读:
    HTTPS原理浅析
    Java8 HashMap源码分析
    Java8 ArrayList源码分析
    Java反射
    Java泛型
    Tensorflow卷积神经网络
    Java8 Stream简介
    java.io与网络通信
    Python实现RNN
    域名系统DNS简介
  • 原文地址:https://www.cnblogs.com/wind-chaser/p/11352788.html
Copyright © 2011-2022 走看看