语义分割 HybridTaskCascade 跑通DEMO 训练自己的数据集

火腿肠项目地址　https://github.com/qixuxiang/Hybrid_Task_Cascade

搭建环境

• Linux (tested on Ubuntu 16.04 and CentOS 7.2)
• Python 3.4+
• PyTorch 1.0
• Cython
• mmcv >= 0.2.2

首先把Pytorch1.0装好

编译

下载项目文件 确保777权限　不然会有编译失败

./compile.sh   

python setup.py develop

或者

python setup.py install

测试

mydemo.py

import mmcv
from mmcv.runner import load_checkpoint
from mmdet.models import build_detector
from mmdet.apis import inference_detector, show_result

cfg = mmcv.Config.fromfile('/media/yang/56c81da5-109b-42e1-9e51-0f9e1396fa9d/yang/Hybrid_Task_Cascade-master/configs/cascade_mask_rcnn_x101_64x4d_fpn_1x.py')
cfg.model.pretrained = None

# construct the model and load checkpoint
model = build_detector(cfg.model, test_cfg=cfg.test_cfg)
_ = load_checkpoint(model, 'checkpoints/cascade_mask_rcnn_x101_64x4d_fpn_20e_20181218-630773a7.pth')

# test a single image
img = mmcv.imread('demo/33823288584_1d21cf0a26_k.jpg')
result = inference_detector(model, img, cfg)
show_result(img, result)
'''
# test a list of images
imgs = ['test1.jpg', 'test2.jpg']
for i, result in enumerate(inference_detector(model, imgs, cfg, device='cuda:0')):
    print(i, imgs[i])
    show_result(imgs[i], result)
'''

权重下载地址　https://github.com/qixuxiang/Hybrid_Task_Cascade/blob/master/MODEL_ZOO.md　

作者试验了很多模型　什么都有　看自己