介绍
该代码用于从学习maskrcnn算法。 我对此做了非常详细的代码注释。 并且我创建了maskrcnn的ros节点来抓取机器人。
当前过程
进行详细的代码注释
写一个maskrcnn的ros节点
编写数据增强python脚本以生成更多数据
文件功能
mrcnn文件:maskrcnn的核心算法
data_aug.py:数据扩充
train_segment.py:模型训练和测试
MaskRCNNTrainGraph.mmap:maskrcnn算法流程图
segment_node.py:maskrcnn推断的ros节点
安装
您应该安装以下库:
numpy scipy Pillow cython matplotlib scikit-image tensorflow>=1.3.0 keras>=2.0.8 opencv-python h5py imgaug IPython
模型
预训练
2021-09-04 15:12:19
962KB
系统开源
1