新增 ResNet34 ImageNet 预训练 + 10% 数据微调模型 (baseline)

baseline/compare_models.py

@@ -22,6 +22,7 @@ from sklearn.metrics import roc_curve, auc, accuracy_score
 from Model import Net
 from Dataloader import RobustImageFolder
 from baseline.VGG_KNN import VGGKNNBaseline
+from baseline.ResNet34_Pretrained_10pct import get_resnet34_10pct_preds
 
 matplotlib.rcParams['font.sans-serif'] = ['SimHei', 'DejaVu Sans']
 matplotlib.rcParams['axes.unicode_minus'] = False
@@ -77,8 +78,9 @@ def get_vgg_knn_preds(train_loader, val_loader, device):
 # ============================================================
 
 MODELS = [
-    ('ResNet-34',           get_resnet34_preds),
-    ('VGG16 + KNN  (K=5)',  get_vgg_knn_preds),
+    ('ResNet-34',                get_resnet34_preds),
+    ('ResNet-34 (10% Fine-tune)', get_resnet34_10pct_preds),
+    ('VGG16 + KNN  (K=5)',       get_vgg_knn_preds),
     # 未来轻松扩展示例：
     # ('ResNet-18 (pretrained)', get_resnet18_preds),
     # ('ResNet-50 (pretrained)', get_resnet50_preds),