微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

尝试访问预测掩码的位置或坐标 Mask-RCNN

如何解决尝试访问预测掩码的位置或坐标 Mask-RCNN

我们正在开发一个视觉控制机器人项目,用于拾取和放置部分被遮挡的物体。我们正在使用掩码 RCNN 架构。到目前为止,我们已经成功地制作了掩码、边界框和获取 bBox 的位置。现在为了有效地处理遮挡,我们需要蒙版的坐标,以便我们可以识别重叠区域。但它不起作用。这是一个代码快照和它产生的结果。

import skimage
import matplotlib.pyplot as plt

real_test_dir = '/content/drive/MyDrive/MaskRCNN_occ/dataset/updated data 2/predict'
image_paths = []
for filename in os.listdir(real_test_dir):
    if os.path.splitext(filename)[1].lower() in ['.png','.jpg','.jpeg']:
        image_paths.append(os.path.join(real_test_dir,filename))

for image_path in image_paths:
    img = skimage.io.imread(image_path)
    
    if img.ndim != 3:
      image = skimage.color.gray2rgb(img)
    if img.shape[-1] == 4:
      image = img[...,:3]

    print(image_path)
    img_arr = np.array(image)
    results = model.detect([img_arr],verbose=1)
    r = results[0]
    print(r['masks'])

它正在产生这些结果,

/content/drive/MyDrive/MaskRCNN_occ/dataset/updated data 2/predict/Image 3.png
Processing 1 images
image                    shape: (256,256,3)         min:    5.00000  max:  255.00000  uint8
molded_images            shape: (1,512,3)      min: -117.70000  max:  151.10000  float64
image_Metas              shape: (1,18)               min:    0.00000  max:  512.00000  float64
anchors                  shape: (1,65472,4)         min:   -0.17712  max:    1.05188  float32
[[[False False False False False]
  [False False False False False]
  [False False False False False]
  ...
  [False False False False False]
  [False False False False False]
  [False False False False False]]]

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?