Mask RCNN 批量预测

源自https://blog.csdn.net/qq_15969343/article/details/80388311

复制那个demo.ipynb

最后一个cell 改为

file_names = next(os.walk(IMAGE_DIR))[2]
for x in range(len(file_names)):
    image = skimage.io.imread(os.path.join(IMAGE_DIR, file_names[x]))
 
# Run detection
    results = model.detect([image], verbose=1)
 
# Visualize results
    r = results[0]
    visualize.display_instances(image, r['rois'], r['masks'], r['class_ids'],
                            class_names, r['scores'])