MaskRCNN
chicken3wings
我要成为神仙程序员儿
展开
-
matterport_MaskRCNN(6)——部分代码功能修改(批量预测+仅显示MaxConfidence)
批量预测(1)先在predict.py所在的文件夹,创建TestResult文件件(2)将predict.py中代码替换成from keras.layers import Inputfrom mask_rcnn import MASK_RCNN from PIL import Imageimport osdef SinglePredict(filename): try: # PIL读取图片 image = Image.open(filename)原创 2020-05-13 14:00:38 · 834 阅读 · 0 评论 -
matterport_MaskRCNN(5)——代码解读model.py
model.py——def load_image_gt()用于载入图片的相关标注信息(),即根据给定的image_id,从dataset中载入:原图,mask图,boxes信息输入:数据集实例,配置类实例,image_id,其他可选参数输出:原图([h,w,3]),图片原始形状,class_ids(图片中各目标的类别id),bbox(各目标的坐标信息),mask(各个目标的掩膜信息)d...原创 2020-01-02 15:02:14 · 1795 阅读 · 3 评论 -
matterport_MaskRCNN(4)——测试模型
注意事项:github的该项目在运行setup.py之后,会生成一个.egg文件,在我的mask虚拟环境中路径是D:\Anaconda\envs\mask\Lib\site-packages\mask_rcnn-2.1-py3.7.egg,当我们要修改一些文件的源码时,需要从该文件修改,并重启编译器才能生效。.egg文件其实是一个压缩包形式,所以我这里右键,使用好压打开,然后使用内部查看器可以需...原创 2019-12-18 14:07:20 · 750 阅读 · 4 评论 -
matterport_MaskRCNN(3)——训练模型
训练代码如下,模型文件会保存在该文件的上级目录的logs文件夹中,每个epoch结束保存一次# -*- coding: utf-8 -*-"""Created on Wed Nov 6 19:06:46 2019@author: Lenovo"""# -*- coding: utf-8 -*-"""Created on Tue Sep 17 15:55:49 2019@a...原创 2019-12-17 15:09:40 · 624 阅读 · 0 评论