e9d5373461b0a037716722a461cb12416aa5d1bc,test.py,,,#,169

Before Change

model_ft = models.resnet50(pretrained=False)
num_ftrs = model_ft.fc.in_features
add_block = []
add_block += [nn.Dropout()]
add_block += [nn.Linear(num_ftrs,256)]
add_block += [nn.BatchNorm1d(256)]
add_block += [nn.LeakyReLU(0.1)]
add_block += [nn.Linear(256, 751)]
model_ft.fc = nn.Sequential(*add_block)
model = load_network(model_ft)

// remove the final fc layer

After Change

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// Load Collected data Trained model
print("-------test-----------")
model_structure = ft_net(751)
model = load_network(model_structure)

// Remove the final fc layer and classifier layer

Instances