Giter VIP home page Giter VIP logo

colab_fasterrcnn_pytorch's Introduction

Colab_FasterRCNN_Pytorch

在Google云盘的Colab跑Faster-Rcnn pytorch版

该代码主要参考了jwyang/faster-rcnn.pytorch的PyTorch复现工程

参考源码:

faster-rcnn pytorch代码下载

一定要下载对应pytorch版本的源码!不然运行版本不兼容

【数据集】

此次训练使用的是VOC 2007数据集

1.下载压缩文件到data

2.解压数据到data/VOCdevkit

  • tar xvf VOCtrainval_06-Nov-2007.tar
  • tar xvf VOCtest_06-Nov-2007.tar
  • tar xvf VOCdevkit_08-Jun-2007.tar

3.创建软链接

  • cd faster-rcnn.pytorch/data

  • ln -s VOCdevkit的绝对路径 VOCdevkit2007

  • Tips:其实这步可以不执行,直接将VOCdevkit文件名改成VOCdevkit2007,PASCAL VOC 2010 and 2012、COCO等数据集也是如此操作。

  • 若使用COCO数据集训练,可参考链接:准备COCO数据集

【下载预训练模型】

下载相应的预训练权重,并放到data/pretrained_model文件夹下,从实验发现caffe得到的预训练权重模型精度更高,因此使用了caffe的预训练权重。

【测试】

本次colab中只包含到训练部分展示,测试可根据具体情况进行

训练完想要测试模型在测试集上的前向效果,运行如下指令:

python test_net.py --dataset pascal_voc --net vgg16 \
                   --checksession $SESSION --checkepoch $EPOCH --checkpoint $CHECKPOINT \
                   --cuda

SESSION、EPOCH、CHECKPOINT修改为自己想要前向测试的模型

操作步骤可参考链接:第8步骤测试test_.net.py

【使用自己数据集】

若想要使用自己的数据集训练,可参考链接第三步使用自己的数据集训练

参考链接

自编写详细教程:[百度网盘word教程](链接:https://pan.baidu.com/s/18tcsezRLBQoQDOfWpi9gFw?pwd=pr62 提取码:pr62)

参考CSDN链接:

参考视频:

colab_fasterrcnn_pytorch's People

Contributors

yaoliu0803 avatar

Watchers

Kostas Georgiou avatar  avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.