zoukankan      html  css  js  c++  java
  • Fast RCNN笔记3:tf-faster-rcnn 使用训练好的fastrcnn模型测试test

    tensorflow版本fastrcnn实现,原github:https://github.com/endernewton/tf-faster-rcnn

    参考:1. https://github.com/CharlesShang/TFFRCNN 

       2. https://blog.csdn.net/char_QwQ/article/details/80980505

    基本可以按照endernewton在git中来做,除了接下来1.没有有写的和2.需要改的部分对应原文中Train your own model

    接下来主要实现训练好的数据对Pascal voc2007数据集进行测试:

    1.将数据集下载完成后解压

    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
    wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
    tar xvf VOCtrainval_06-Nov-2007.tar
    tar xvf VOCtest_06-Nov-2007.tar
    tar xvf VOCdevkit_08-Jun-2007.tar

    将文件夹以VOCdevkit2007命名copy到data下,形成以下目录结构/tf-faster-rcnn/data/VOCdevkit2007/

    其下应该有以下文件(也可以参照CharlesShang中用软链接):

    2.如果完成了第一步demo测试那么就可以直接运行test(没有需要下载pre-trained model参照原git中

    Train your own model

    1.Download pre-trained models and weights. ):

    运行之前还需要修改tf-faster-rcnn/lib/datasets/voc_eval.py下105行和121行。

    105:将

    cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile)

    注释掉改为

    cachefile = os.path.join(cachedir, '%s_annots.pkl' % imagesetfile.split("/")[-1].split(".")[0])

    121行w改为wb

      

    运行测试命令:

    GPU_ID=0
    ./experiments/scripts/test_faster_rcnn.sh $GPU_ID pascal_voc_0712 res101

    GPU_ID自己看用哪一块,res101:下载的model pascal是数据集格式

    3.运行结果如下:

  • 相关阅读:
    Linux运维相关命令
    Linux常用命令合集
    python爬虫
    Linux常见知识点
    mysql linux上安装使用
    非托管代码方面的问题
    有趣知识
    C# 模式匹配
    C# 泛型约束
    怎么查看自己电脑的IP地址?
  • 原文地址:https://www.cnblogs.com/millerfu/p/9682237.html
Copyright © 2011-2022 走看看