代码仓库地址
https://github.com/Megvii-BaseDetection/BorderDet/tree/bade7dfdd2e8f25e492d89209534ea243793919e
下载代码压缩包,解压
如下图所示
空白处点击鼠标右键,点击open in terminal
创建虚拟环境conda create -n BorderDet python=3.6
激活虚拟环境 conda activate BorderDet
install cvpods locally
官方代码
python3 -m pip install 'git+https://github.com/Megvii-BaseDetection/cvpods.git'
# (add --user if you don't have permission)
# Or, to install it from a local clone:
git clone https://github.com/Megvii-BaseDetection/cvpods.git
python3 -m pip install -e cvpods
# Or,
pip install -r requirements.txt
python3 setup.py build develop
我先运行的第三种,但是会报错,这里直接说第二种的步骤。
命令行输入git clone https://github.com/Megvii-BaseDetection/cvpods.git
可能会报一个跟git有关的错,只需要安装git即可,输入conda install git
,等待安装完成即可
在运行下一条python3 -m pip install -e cvpods
之前,需要先安装torch,不然会报错ModuleNotFoundError:No module named 'torch'
安装torch:先查看本机的cuda版本,命令行输入nvcc -V
,可以查看,查看后到pytorch官网找对应的pytorch版本:
最后一行release后面就是版本,我的是10.1
去这个网站:https://pytorch.org/get-started/previous-versions/找对应的版本即可
复制对应的版本那行代码到命令行,如图
输入python3 -m pip install -e cvpods
,会出现很多行,后来又会报错error: command 'gcc' failed with exit status 1
,如图所示
经过好一阵百度,找到了解决方法
输入sudo apt-get install nvidia-cuda-toolkit
,然后输入密码会生成好多东西,不截图了
输入pip install -e cvpods --user
,又是很长一串,
prepare datasets
官方给的代码
cd /path/to/cvpods
cd datasets
ln -s /path/to/your/coco/dataset coco
先把数据集准备好:按照cvpods/datasets里面的README.md里的格式
然后将地址链接到对应的位置,注意地址要用你自己的,不要粘贴我的
输入cd cvpods/datasets
输入ln -s /home/jxz/NewDisk/UbuntuCode/dataset/coco coco
train & test
??又报错了