查看: 2453|回复: 0

[AI源码] maskscoringRCNN_pytorch源码系统

[复制链接]

166

主题

616

帖子

1万

积分

xdtech

Rank: 5Rank: 5

积分
11585
发表于 2019-4-2 09:42:21 | 显示全部楼层 |阅读模式
本帖最后由 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. 以上各条及其他未尽事宜,商家保留最终解释权。



回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表