配置facebook的maskrcnn-benchmark

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/mdjxy63/article/details/89298070

repo:https://github.com/facebookresearch/maskrcnn-benchmark

安装:https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/INSTALL.md

可以按照这个教程安装,不过需要注意几点问题:

1)一定要使用anaconda的虚拟环境

2)一定要先安装Ipython

3)不能按照教程那样使用conda安装pytorch,因为现在使用conda安装 pytorch已经是pytorch1.1的版本,此时的model_zoo的函数已经修改了,下面这三行会报错。因此使用pip安装,pip install torch==1.0.0 torchvision

from torch.utils.model_zoo import _download_url_to_file
from torch.utils.model_zoo import urlparse
from torch.utils.model_zoo import HASH_REGEX

4)因为pycharm路径的原因,所以导致测试coco数据集的时候会找不到json文件,因此,解决办法是,指定全路径

在paths_catalog.py里面,修改DATA_DIR变量

猜你喜欢

转载自blog.csdn.net/mdjxy63/article/details/89298070