Darknet-19 pretrained weight 에서 input normalization 문제.

pretrained weight를 가져와서 테스트해보니 input을 zero-mean으로 하면 안되고 그냥 그대로(?) to_tensor로 만들어야 정답이 제대로 나옵니다.

가령 아래와 같이 짜면 값이 안 나오고

dataset = dset.ImageFolder(root="samples/",
                           transform=transforms.Compose([
                               transforms.Resize((256,256)),
                               transforms.ToTensor(),       # Tensor로 바꾸고 (0~1로 자동으로 normalize)
                               transforms.Normalize(mean=[0.485, 0.456, 0.406],
                                                     std=[0.229, 0.224, 0.225]),
                           ]))

이렇게 해야합니다..

dataset = dset.ImageFolder(root="samples/",
                           transform=transforms.Compose([
                               transforms.Resize((256,256)),
                               transforms.ToTensor()      # Tensor로 바꾸고 (0~1로 자동으로 normalize)
                           ]))

DeepBaksuVision / YOLO9000

Darknet-19 pretrained weight 에서 input normalization 문제. #33