一、ChainerCV
ChainerCV,一个用于计算机视觉的深度学习实用库。这个库旨在通过 Chainer 简易化计算机视觉的训练和深度学习模型应用的过程。它包含计算机视觉模型的高质量实现,以及开展计算机视觉研究的必备工具集。当前,ChainerCV 提供了目标检测和语义分割模型(Faster R-CNN、SSD 和 SegNet)的实现。
https://github.com/chainer/chainercv
二、修改原始代码
1,chainercv/datasets/voc/voc_utils.py中
(1)更改voc_bbox_label_names,改为自己数据库的类标;增加oc_semantic_segmentation_label_colors,和类标相对应