Skip to content

Hrnet训练过程中发现一个小问题 #855

@Surfeey

Description

@Surfeey

加载数据集的文件my_dataset_coco.py里,CocoKeypoint的
def getitem(self, idx):
target = copy.deepcopy(self.valid_person_list[idx])

    image = cv2.imread(target["image_path"])
    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
    if self.transforms is not None:
        image, person_info = self.transforms(image, target)

    return image, target

上述代码return的应该是person_info而不是target吧,我训练用的是代码库里的train.py,调用时一般大家都会输入transforms吧。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions