Jiao77/RoRD-Layout-Recognation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

turn vgg from pretrained to random

Browse Source
This commit is contained in:
Jiao77
2025-06-09 00:55:28 +08:00
parent f0b2e1b605
commit 9042815b34
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/rord.py
View File

@@ -13,7 +13,7 @@ class RoRD(nn.Module):
""" """
super(RoRD, self).__init__() super(RoRD, self).__init__()
vgg16_features = models.vgg16(pretrained=True).features vgg16_features = models.vgg16(pretrained=False).features
# 共享骨干网络 # 共享骨干网络
self.slice1 = vgg16_features[:23] # 到 relu4_3 self.slice1 = vgg16_features[:23] # 到 relu4_3