程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2024-11(2)

【Faster R-CNN实现】基于pytorch-faster-rcnn的目标检测完整流程

发布于2019-08-07 12:18     阅读(3924)     评论(0)     点赞(3)     收藏(4)


参考链接:pytorch-faster-rcnn


第一步:安装Anaconda,参考【Ubuntu】安装Anaconda并搭建虚拟环境和迁移

第二步:下载pytorch及其他安装包(注意以下所有操作都将在虚拟环境pytorch下进行)

  • 下载pytorch(CUDA版本):https://pytorch.org/                                                                                                             
    1. conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
    2. # or use pip
    3. pip3 install torch torchvision
  • 下载opencv-python, easydict及matplotlib,scipy:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python, easydict==1.6, matplotlib, tensorboardX, scipy==1.2, PyYAML

第三步:安装pytorch-faster-rcnn及数据接口

  注意:若不是在虚拟环境下,Python COCO API将不能使用

第四步:准备数据集并链接(下载PASCAL VOC 2007

  注意:在第4步中,完整命令应该是:

(pytorch) zzw@amax:/data1/zzw/Pytorch-test/pytorch-faster-rcnn/data$ ln -s /data1/zzw/Datasets/VOCdevkit VOCdevkit2007

  以上显示是Datasets数据集的路径,及数据集链接成功后的结果

第五步:下载预训练模型并链接

  • Google drive here.

  比如下载res101模型并解压,将res101_faster_rcnn_iter_110000.pth放到下面的目录  

 pytorch-faster-rcnn/data/voc_2007_trainval+voc_2012_trainval

  接下来,在pytorch-faster-rcnn目录下,使用以下命令创建一个output及其子文件

  1. NET=res101
  2. TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
  3. mkdir -p output/${NET}/${TRAIN_IMDB}
  4. cd output/${NET}/${TRAIN_IMDB}
  5. ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
  6. cd ../../..

第六步:测试预训练模型

 



所属网站分类: 技术文章 > 博客

作者:大壮

链接:https://www.pythonheidong.com/blog/article/10726/7f74b7c1b7a2ca0a99d9/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

3 0
收藏该文
已收藏

评论内容:(最多支持255个字符)