Skip to content
This repository was archived by the owner on Oct 31, 2023. It is now read-only.

Commit cbf4a09

Browse files
committed
refine densepose coco evaluation code
1 parent 35e69d1 commit cbf4a09

File tree

4 files changed

+274
-287
lines changed

4 files changed

+274
-287
lines changed

detectron/core/test_engine.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -374,7 +374,7 @@ def get_roidb_and_dataset(dataset_name, proposal_file, ind_range):
374374

375375

376376
def empty_results(num_classes, num_images):
377-
"""Return empty results lists for boxes, masks, and keypoints.
377+
"""Return empty results lists for boxes, masks, keypoints and body IUVs.
378378
Box detections are collected into:
379379
all_boxes[cls][image] = N x 5 array with columns (x1, y1, x2, y2, score)
380380
Instance mask predictions are collected into:
@@ -387,7 +387,10 @@ def empty_results(num_classes, num_images):
387387
Keypoints are recorded for person (cls = 1); they are in 1:1
388388
correspondence with the boxes in all_boxes[cls][image].
389389
Body uv predictions are collected into:
390-
TODO
390+
all_bodys[cls][image] = [...] list of body IUV results that are in 1:1
391+
correspondence with the boxes in all_boxes['person'][image], each encoded
392+
as a 3D array (3, int(bbox_height), int(bbox_width)) with the 3 rows
393+
corresponding to [Index, U, V].
391394
"""
392395
# Note: do not be tempted to use [[] * N], which gives N references to the
393396
# *same* empty list.

0 commit comments

Comments
 (0)