lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

训练数据前置处理与提升训练效率

Browse Source
This commit is contained in:
lee
2025-07-10 14:24:05 +08:00
parent 0701538a73
commit 09f41f6289
15 changed files with 430 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
model/resnet_pre.py
View File

@ -222,13 +222,13 @@ class ResNet(nn.Module):
self.bn1 = norm_layer(self.inplanes)
self.relu = nn.ReLU(inplace=True)
self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2, padding=1)
self.adaptiveMaxPool = nn.AdaptiveMaxPool2d((1, 1))
self.maxpool2 = nn.Sequential(
nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
nn.MaxPool2d(kernel_size=2, stride=1, padding=0)
)
# self.adaptiveMaxPool = nn.AdaptiveMaxPool2d((1, 1))
# self.maxpool2 = nn.Sequential(
# nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
# nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
# nn.MaxPool2d(kernel_size=2, stride=2, padding=0),
# nn.MaxPool2d(kernel_size=2, stride=1, padding=0)
# )
self.layer1 = self._make_layer(block, int(64 * scale), layers[0])
self.layer2 = self._make_layer(block, int(128 * scale), layers[1], stride=2,
dilate=replace_stride_with_dilation[0])