In SetCriterionHOI__init__, how the initial object (self.obj_nums_init) and verb (self.verb_nums_init) initial numbers are set? There two arrays for HICO-DET and V-COCO are hard coded here.
Are these just counts over categories from ground truth annotations? I performed an count for the verb categories using the following code snippet. This computation is done at the end of HICODetection__init__here.
from collections import Counter
hoi_annotations = [
ann['hoi_category_id']
for annotation in self.annotations
for ann in annotation['hoi_annotation']
]
hoi_annotations_valid = {
k: v
for k, v in Counter(hoi_annotations).items()
if k in self._valid_verb_ids
}
verb_nums = [(k, hoi_annotations_valid[k]) for k in self._valid_verb_ids]
In
SetCriterionHOI
__init__
, how the initial object (self.obj_nums_init
) and verb (self.verb_nums_init
) initial numbers are set? There two arrays for HICO-DET and V-COCO are hard coded here.Are these just counts over categories from ground truth annotations? I performed an count for the verb categories using the following code snippet. This computation is done at the end of
HICODetection
__init__
here.As shown above, the obtained
verb_nums
did not matchself.verb_nums_init
here.