Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
曾经的那一瞬间
Models
提交
00f71bf9
M
Models
项目概览
曾经的那一瞬间
/
Models
11 个月 前同步成功
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
Models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
00f71bf9
编写于
12月 20, 2021
作者:
A
A. Unique TensorFlower
提交者:
TF Object Detection Team
12月 20, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Image-level labels are not propagated to the open images challenge metric.
PiperOrigin-RevId: 417394640
上级
48b4b573
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
34 addition
and
1 deletion
+34
-1
research/object_detection/core/model.py
research/object_detection/core/model.py
+8
-0
research/object_detection/inputs.py
research/object_detection/inputs.py
+2
-1
research/object_detection/model_lib.py
research/object_detection/model_lib.py
+21
-0
research/object_detection/utils/object_detection_evaluation.py
...rch/object_detection/utils/object_detection_evaluation.py
+3
-0
未找到文件。
research/object_detection/core/model.py
浏览文件 @
00f71bf9
...
...
@@ -327,6 +327,7 @@ class DetectionModel(six.with_metaclass(abc.ABCMeta, _BaseClass)):
groundtruth_not_exhaustive_classes
=
None
,
groundtruth_keypoint_depths_list
=
None
,
groundtruth_keypoint_depth_weights_list
=
None
,
groundtruth_image_classes
=
None
,
training_step
=
None
):
"""Provide groundtruth tensors.
...
...
@@ -398,6 +399,9 @@ class DetectionModel(six.with_metaclass(abc.ABCMeta, _BaseClass)):
groundtruth_keypoint_depth_weights_list: a list of 2-D tf.float32 tensors
of shape [num_boxes, num_keypoints] containing the weights of the
relative depths.
groundtruth_image_classes: A list of 1-D tf.float32 tensors of shape
[num_classes], containing label indices encoded as k-hot of the classes
that are present or not present in the image.
training_step: An integer denoting the current training step. This is
useful when models want to anneal loss terms.
"""
...
...
@@ -474,6 +478,10 @@ class DetectionModel(six.with_metaclass(abc.ABCMeta, _BaseClass)):
self
.
_groundtruth_lists
[
fields
.
InputDataFields
.
groundtruth_verified_neg_classes
]
=
groundtruth_verified_neg_classes
if
groundtruth_image_classes
:
self
.
_groundtruth_lists
[
fields
.
InputDataFields
.
groundtruth_image_classes
]
=
groundtruth_image_classes
if
groundtruth_not_exhaustive_classes
:
self
.
_groundtruth_lists
[
fields
.
InputDataFields
...
...
research/object_detection/inputs.py
浏览文件 @
00f71bf9
...
...
@@ -668,7 +668,8 @@ def _get_labels_dict(input_dict):
fields
.
InputDataFields
.
groundtruth_dp_surface_coords
,
fields
.
InputDataFields
.
groundtruth_track_ids
,
fields
.
InputDataFields
.
groundtruth_verified_neg_classes
,
fields
.
InputDataFields
.
groundtruth_not_exhaustive_classes
fields
.
InputDataFields
.
groundtruth_not_exhaustive_classes
,
fields
.
InputDataFields
.
groundtruth_image_classes
,
]
for
key
in
optional_label_keys
:
...
...
research/object_detection/model_lib.py
浏览文件 @
00f71bf9
...
...
@@ -114,6 +114,10 @@ def _prepare_groundtruth_for_eval(detection_model, class_agnostic,
'groundtruth_not_exhaustive_classes': [batch_size, num_classes] K-hot
representation of 1-indexed classes which don't have all of their
instances marked exhaustively.
'input_data_fields.groundtruth_image_classes': integer representation of
the classes that were sent for verification for a given image. Note that
this field does not support batching as the number of classes can be
variable.
class_agnostic: Boolean indicating whether detections are class agnostic.
"""
input_data_fields
=
fields
.
InputDataFields
()
...
...
@@ -136,6 +140,18 @@ def _prepare_groundtruth_for_eval(detection_model, class_agnostic,
input_data_fields
.
groundtruth_classes
:
groundtruth_classes
}
if
detection_model
.
groundtruth_has_field
(
input_data_fields
.
groundtruth_image_classes
):
groundtruth_image_classes_k_hot
=
tf
.
stack
(
detection_model
.
groundtruth_lists
(
input_data_fields
.
groundtruth_image_classes
))
# We do not add label_id_offset here because it was not added when encoding
# groundtruth_image_classes.
groundtruth_image_classes
=
tf
.
expand_dims
(
tf
.
where
(
groundtruth_image_classes_k_hot
>
0
)[:,
1
],
0
)
groundtruth
[
input_data_fields
.
groundtruth_image_classes
]
=
groundtruth_image_classes
if
detection_model
.
groundtruth_has_field
(
fields
.
BoxListFields
.
masks
):
groundtruth
[
input_data_fields
.
groundtruth_instance_masks
]
=
tf
.
stack
(
detection_model
.
groundtruth_lists
(
fields
.
BoxListFields
.
masks
))
...
...
@@ -384,6 +400,10 @@ def provide_groundtruth(model, labels, training_step=None):
if
fields
.
InputDataFields
.
groundtruth_not_exhaustive_classes
in
labels
:
gt_not_exhaustive_classes
=
labels
[
fields
.
InputDataFields
.
groundtruth_not_exhaustive_classes
]
groundtruth_image_classes
=
None
if
fields
.
InputDataFields
.
groundtruth_image_classes
in
labels
:
groundtruth_image_classes
=
labels
[
fields
.
InputDataFields
.
groundtruth_image_classes
]
model
.
provide_groundtruth
(
groundtruth_boxes_list
=
gt_boxes_list
,
groundtruth_classes_list
=
gt_classes_list
,
...
...
@@ -405,6 +425,7 @@ def provide_groundtruth(model, labels, training_step=None):
groundtruth_not_exhaustive_classes
=
gt_not_exhaustive_classes
,
groundtruth_keypoint_depths_list
=
gt_keypoint_depths_list
,
groundtruth_keypoint_depth_weights_list
=
gt_keypoint_depth_weights_list
,
groundtruth_image_classes
=
groundtruth_image_classes
,
training_step
=
training_step
)
...
...
research/object_detection/utils/object_detection_evaluation.py
浏览文件 @
00f71bf9
...
...
@@ -870,6 +870,9 @@ class OpenImagesChallengeEvaluator(OpenImagesDetectionEvaluator):
image_classes
=
groundtruth_dict
[
input_fields
.
groundtruth_image_classes
]
elif
input_fields
.
groundtruth_labeled_classes
in
groundtruth_dict
:
image_classes
=
groundtruth_dict
[
input_fields
.
groundtruth_labeled_classes
]
else
:
logging
.
warning
(
'No image classes field found for image with id %s!'
,
image_id
)
image_classes
-=
self
.
_label_id_offset
self
.
_evaluatable_labels
[
image_id
]
=
np
.
unique
(
np
.
concatenate
((
image_classes
,
groundtruth_classes
)))
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录