ImageNet 원본 이미지 보는 방법

이미지넷 원본이 보고싶다면....

일단 다운부터 받자.

 

[논문 구현] ViT ImageNet 평가 방법 | pytorch, timm 라이브러리, timm ViT

 

다음으로 데이터 로더를 설계해준다.

 

transform_origin = transforms.Compose([
    transforms.Resize(256),
    transforms.CenterCrop(224),
    transforms.ToTensor(),
])

origin_set = datasets.ImageFolder('./data/ImageNet/val', transform=transform_origin)
origin_loader = DataLoader(origin_set, batch_size=BATCH_SIZE, shuffle=False, num_workers=NUM_WORKERS)

 

이제 이 메서드를 실행시키면 끝이다.

n번째 사진을 보려고 argument를 받아온다.

 

def show_origin(n):
    for i, data in enumerate(origin_loader):
        if i == n:
            inputs, labels = data
            inputs_np, labels_np = inputs.numpy(), labels.numpy()

            inputs_np = np.transpose(inputs_np, (0, 2, 3, 1))[0]
            plt.imshow(inputs_np)
            plt.title(imagenet_ind2str(int(labels_np)))
            plt.show()
            break

 

결과

이미지 1

이미지 1

끝.