Open superhero-7 opened 3 years ago
在resnet34中最后用于分类的全连接实现是这样的: self.fc = nn.Linear(512, num_classes) 想问一下这里的输入特征维数512应该是算出来的吧,受输入图片大小影响吧?
self.fc = nn.Linear(512, num_classes)
正解,所以输入图片时要resize
thank you! I also find this problem and solve it by your idea.
在resnet34中最后用于分类的全连接实现是这样的:
self.fc = nn.Linear(512, num_classes)
想问一下这里的输入特征维数512应该是算出来的吧,受输入图片大小影响吧?