问题标签 [detectron]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
2 回答
1556 浏览

python-3.x - 如何计算 Detectron2 中的交集?

我正在使用 Detectron2 进行对象检测。我已经注册了 pascalvoc 数据集并训练了一个检测模型。如何计算我的测试数据集的平均 IOU?我知道detection2有一个用于计算IOU的预定义函数,即detectron2.structures.pairwise_iou

我在 csv 文件中有测试图像的地面实况边界框。csv 文件包含(文件名、宽度、高度、类、xmin、ymin、xmax、ymax)。如何解析 IOU 函数中的两个边界框并将其显示在 google colab 中。

这是我生成预测边界框的代码

0 投票
1 回答
848 浏览

python - AttributeError:在给定的实例中找不到字段“pred_masks”

我在 colab 上使用 RetinaNet Detectron2 训练了一个模型。培训成功,但我无法在图像上进行测试。

代码:

错误:

/content/drive/My Drive/Detectron/test/wh_dresscode_4969_jpg.rf.5a2702518691d3d6f9cef7ad7977e5c8.jpg 数据类型:<class 'detectron2.structures.instances.Instances'> AttributeError Traceback(最近一次调用最后一次)

AttributeError:在给定的实例中找不到字段“pred_masks”!

0 投票
1 回答
2308 浏览

json - 如何注册要与detectron2 一起使用的数据集?我们有 COCO JSON 格式的图像及其注释

我正在尝试使用 Detectron2 训练模型。我正在使用杂货店图像数据,并且有 COCO 格式的注释。我在加载模型时遇到问题。模型不带注释。我指的是这个博客https://gilberttanner.com/blog/detectron2-train-a-instance-segmentation-model

注册数据集时面临问题。

这段代码有问题吗?

0 投票
2 回答
3260 浏览

python - 在detectron2中,如何在对象检测中仅在没有标签(类名)的图像上绘制边界?

我正在学习detectron2并使用pokemonster数据练习它。所以,我按照detectron2教程(这里:https ://colab.research.google.com/drive/16jcaJoc6bCFAQ96jDe2HwtXj7BMD_-m5 )

并且还开发了一个检测器,可以检测到下面的图像。

在此处输入图像描述

但是,在这张图片中,红色圆圈是类名。我只想绘制边界框(绿色),红色圆圈(类名)除外。现在下面是我的可视化代码。应该修改什么代码?谢谢你。

0 投票
1 回答
549 浏览

computer-vision - 给 randomCrop 增加和损失变得爆炸

我正在尝试测试不同类型的增强,
但是当我给出选项时,RandomCrop它给出了损失值 NaN 或无穷大。

这是我的随机增强优化

  1. 此代码是否会将增强随机应用于任何输入批处理图像

  2. 为什么当我给予时它会爆炸损失RandomCrop

    FloatingPointError:预测框或分数包含 Inf/NaN。训练有分歧。

0 投票
0 回答
346 浏览

python-3.x - 从 Detectron2 中提取对象特征

我对物体检测真的很陌生,如果这个问题看起来太明显了,很抱歉。

我在检测器 2 上有一个训练有素的 FasterRCNN 模型来检测物体,我正在尝试提取每个检测到的物体的特征,以作为模型预测的输出。可用的教程似乎在寻找投资回报率并做出新的预测,以及它们的框和功能。我有推理的盒子,我只需要提取每个盒子的特征。我在下面添加了我一直在使用的代码。谢谢

0 投票
3 回答
2141 浏览

detectron - Detectron2 检查点未找到

从昨晚开始,我就遇到了这样的错误。我训练了5个模型,没有问题。然后就出现了这样的问题。如果您能提供帮助,我会很高兴。

0 投票
0 回答
580 浏览

python - RuntimeError:将 np 数组转换为列表后,“upsample_bilinear2d_channels_last”未实现“Long”

我正在尝试跨模块发送图像数据。

以下工作正常:

pg_image当作为我的 Detectron2 模型的输入时,此函数 ( ) 的结果可以正常工作。

但是,当我pg_image.tolist()在接收 () 时发送并转换回 np 数组np.array(pg_image)并将其发送到我的 Detectron2 模型时,我不断收到以下错误:

那么,如果将 np 数组转换为列表并转换回更改数据?

0 投票
2 回答
1117 浏览

python - 检测器 2 输出的类标签指的是什么?

在 的文档中detectron2,它指出类标签位于output_dict['Instances'].pred_classes. 这一切都很好,我可以轻松访问它,但在文档(或输出字典,据我所知)中没有任何地方指定哪个整数标签引用哪个类。我假设某处有一本包含类似内容的字典{0: 'Person', 1: 'Bicycle', 2: 'Car',... },但我似乎找不到它。有谁知道在哪里可以找到这个?

以下是输出字典的规格:https ://detectron2.readthedocs.io/tutorials/models.html

0 投票
1 回答
568 浏览

python - 在 Docker 上安装 detectron2 包

我正在使用 python 3.7.9 版本。dectron2 未正确安装。命令:docker exec -it 1ee02b74a599 /bin/bash root@1ee02b74a599:/app# pip install detectron2==0.2.1 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.4/index。 html

安装过程中出现以下问题。