my_road.py
改写mask_rcnn的数据集类,从而可以读取png格式的mask。主要重写了其中的load_mask函数,dataset类。完成从png格式的mask到数据集numpy数组的变换。
下载地址
用户评论