new train

2026-07-28 18:06:10 +08:00
parent 44b0d6b756
commit a55d449028
13 changed files with 224 additions and 27 deletions
@@ -23,7 +23,8 @@ train_transform = transforms.Compose([
    transforms.RandomRotation(10),  # 轻微旋转
    transforms.ColorJitter(brightness=0.1, contrast=0.1),
    transforms.ToTensor(),
-    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
+    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
+    transforms.RandomErasing(p=0.5, scale=(0.02, 0.2), ratio=(0.3, 3.3)),
 ])