SSD - VGG backbone feature
VGG16 feature extract 할 때 feature shape 및 추출 부분 메모 input : [batch, 3, 300, 300] ModuleList( (0): Conv2d(3, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) # torch.Size([32, 64, 300, 300]) (1): ReLU(inplace=True) # torch.Size([32, 64, 300, 300]) (2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) # torch.Size([32, 64, 300, 300]) (3): ReLU(inplace=True) # torch.Size([32, ..
2022. 6. 23.
PASCAL VOC class 순서
detection 코드 정리하다가 pascal voc 클래스 dictionary를 따로 메모해 놓음 {'__background__': 0, 'aeroplane': 1, 'bicycle': 2, 'bird': 3, 'boat': 4, 'bottle': 5, 'bus': 6, 'car': 7, 'cat': 8, 'chair': 9, 'cow': 10, 'diningtable': 11, 'dog': 12, 'horse': 13, 'motorbike': 14, 'person': 15, 'pottedplant': 16, 'sheep': 17, 'sofa': 18, 'train': 19, 'tvmonitor': 20}
2022. 6. 22.
