deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard

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

Open Surfeey opened this issue 6 months ago • 0 comments

加载数据集的文件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吧。

Surfeey avatar May 26 '25 18:05 Surfeey