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.