美文网首页
DL notes[2]: faster RCNN with py

DL notes[2]: faster RCNN with py

作者: isSen | 来源:发表于2018-10-17 21:41 被阅读0次

    1. github source

    https://github.com/jwyang/faster-rcnn.pytorch
    下载到本机
    git clone https://github.com/jwyang/faster-rcnn.pytorch.git
    cd faster-rcnn.pytorch && mkdir data
    安装依赖项
    pip install -r requirements.txt
    编译cuda依赖
    cd lib
    sh make.sh

    2. push to the server

    将本机编译好的文件上传服务器
    服务器缺少一些项目造成的额外操作
    apt update
    apt install libglib2.0-0
    apt-get install libsm6 libxrender1 libfontconfig1
    apt install libxext6

    3. run the trainval.sh

    创建一个sh文件

    # trainval_sh.sh content
    CUDA_VISIBLE_DEVICES=0 python trainval_net.py --dataset pascal_voc \
    --net vgg16 --bs 2 --nw 2 --lr 4e-3 \
    lr_decay_step 10 --cuda
    

    运行sh文件
    sh trainval_sh.sh

    相关文章

      网友评论

          本文标题:DL notes[2]: faster RCNN with py

          本文链接:https://www.haomeiwen.com/subject/xfldzftx.html