matterport_MaskRCNN（6）——部分代码功能修改（批量预测+仅显示MaxConfidence）

最新推荐文章于 2022-07-11 16:17:43 发布

chicken3wings

最新推荐文章于 2022-07-11 16:17:43 发布

阅读量849

点赞数

分类专栏： MaskRCNN

本文链接：https://blog.csdn.net/chicken3wings/article/details/106096553

版权

本博客介绍了如何修改matterport_MaskRCNN项目的代码，实现批量预测功能，并在绘图时仅显示具有最高置信度的mask。通过在predict.py中创建TestResult文件夹，更新预测代码，以及在visualize.py中添加保存图片的语句和修改显示mask的逻辑，可以达到预期效果。

摘要由CSDN通过智能技术生成

批量预测

（1）先在predict.py所在的文件夹，创建TestResult文件件
（2）将predict.py中代码替换成

from keras.layers import Input
from mask_rcnn import MASK_RCNN 
from PIL import Image

import os

def SinglePredict(filename):
    try:
    	# PIL读取图片
        image = Image.open(filename)
    except:
        print('Open Error! Try again!')
    else:
        # 预测并保存图片
        mask_rcnn.detect_image(image)
        print(