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
网友评论