|
本帖最后由 Happy清子 于 2019-4-2 09:51 编辑
名称
maskscoringRCNN_pytorch源码
功能介绍
本代码为Pytorch版本的MaskScoring RCNN,特征网采用ResNet50。训练数据集为COCO 2014格式。
Github源码地址
https://github.com/zjhuang22/maskscoring_rcnn
使用方法(demo)
1. 解压后,代码在maskscoring_rcnn-master目录。
2. 在该目录同级处,建立目录pretrained_models. 将训练好的权值放入。
3. 在maskscoring_rcnn-master目录,运行:
python ./tools/demo.py
使用方法(train)
1. 解压后,代码在maskscoring_rcnn-master目录。
2. 在该目录同级处,建立目录pretrained_models。将预训练权值拷入。
3. 安装必备包
pip install ninja yacs cython matplotlib
注意:平台环境,cython等已安装。但此处还是要再安装一次。
4. 替换pth文件。
将maskscoring_rcnn-master目录的easy-install.pth覆盖/home/ubuntu/MyFiles/.local的同名文件。
5. cocoapi安装,在cocoapi/PythonAPI目录:python setup.py build_ext install --prefix /home/ubuntu/MyFiles/.local
6. 安装工程,在maskscoring_rcnn-master目录:
python setup.py build develop --prefix /home/ubuntu/MyFiles/.local
7. 运行方式:
Pycharm,运行/tools/train_net.py开始训练。
系统环境说明
已在平台的Pytorch1.0,测试通过。
基础参数:Python3.6,Pytorch1.0。
推荐配置
慢速训练:单1080Ti显卡,4核CPU,10G内存。
快速训练:多1080ti显卡,8核CPU,20G内存。
免费服务承诺
交付完成的标志是:买家建立Pytorch1.0环境的主机,商家将代码在该主机中调通。
调通的标准是:demo可用作者提供的权值文件进行目标检测(COCO2014目标类型),train可在COCO2014数据集上正常训练。
运行方式:在Pycharm调通。
注意事项
1. 商家仅承诺在平台主机(Pytorch1.0环境),其他环境下本代码的调通,属于额外的增值服务,可能需要另外收费。其他环境包括:买家自己的电脑,平台的其他环境如conda_DL、DL全家桶等。
2. 在买家自定义数据集上跑通本代码,属于额外的增值服务,可能需要另外收费。
3. 买家在购买本代码后,又有进一步的开发需求,开发费用可享受8-9折优惠,具体优惠额再行商定。
4. 以上各条及其他未尽事宜,商家保留最终解释权。
|
|