Wooks_learning

현재 3가지 object에 대해 학습을 진행하고 있는데 모델에 들어가는 annotation이 transform에 맞지않게 들어가 학습되는 문제를 겪어 시간을 허비했다. classification은 transform에 따라 label이 변하지않아 따로 확인을 안했었는데 어떤것이든 학습 전에 input data들을 살펴보는 것이 중요할 것 같다. 아래 코드는 학습 데이터(batch_size = 16)를 trainloader에서 img, annotation, class_index를 받아와 이미지와 각각 바운딩 박스를 플롯해주는 코드이다. input size를 고려해 cnt에 더해주는 값을 넣어주면 된다. def matplotlib_imshow(img, anno, idx, one_channel=False): f..