ssd2 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. SSD(single shot detector) 코드 분석 1. (data) 아래 SSD 코드를 보고 필요한 내용 개인적으로 기록 https://github.com/lufficc/SSD GitHub - lufficc/SSD: High quality, fast, modular reference implementation of SSD in PyTorch High quality, fast, modular reference implementation of SSD in PyTorch - GitHub - lufficc/SSD: High quality, fast, modular reference implementation of SSD in PyTorch github.com PASCAL VOC data training data : PASCAL VOC 2007 & 2012 testing data.. 2022. 6. 4. 이전 1 다음