发布于2019-08-07 12:18 阅读(3924) 评论(0) 点赞(3) 收藏(4)
参考链接:pytorch-faster-rcnn
第一步:安装Anaconda,参考【Ubuntu】安装Anaconda并搭建虚拟环境和迁移
第二步:下载pytorch及其他安装包(注意以下所有操作都将在虚拟环境pytorch下进行)
- conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
-
- # or use pip
- pip3 install torch torchvision
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数据集的路径,及数据集链接成功后的结果
第五步:下载预训练模型并链接
比如下载res101模型并解压,将res101_faster_rcnn_iter_110000.pth放到下面的目录
pytorch-faster-rcnn/data/voc_2007_trainval+voc_2012_trainval
接下来,在pytorch-faster-rcnn目录下,使用以下命令创建一个output及其子文件
- NET=res101
- TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
- mkdir -p output/${NET}/${TRAIN_IMDB}
- cd output/${NET}/${TRAIN_IMDB}
- ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
- cd ../../..
第六步:测试预训练模型
作者:大壮
链接:https://www.pythonheidong.com/blog/article/10726/7f74b7c1b7a2ca0a99d9/
来源:python黑洞网
任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任
昵称:
评论内容:(最多支持255个字符)
---无人问津也好,技不如人也罢,你都要试着安静下来,去做自己该做的事,而不是让内心的烦躁、焦虑,坏掉你本来就不多的热情和定力
Copyright © 2018-2021 python黑洞网 All Rights Reserved 版权所有,并保留所有权利。 京ICP备18063182号-1
投诉与举报,广告合作请联系vgs_info@163.com或QQ3083709327
免责声明:网站文章均由用户上传,仅供读者学习交流使用,禁止用做商业用途。若文章涉及色情,反动,侵权等违法信息,请向我们举报,一经核实我们会立即删除!