first commit

fine/fineConfig.py

@@ -2,7 +2,7 @@ import os
 import pandas as pd
 import torch
 
-from fine.transforms import transforms
+from torchvision import  transforms
 
 
 
@@ -14,9 +14,7 @@ def load_data_transformers(resize_reso=512, crop_reso=448, swap_num=[7, 7]):
     center_resize = 600
     Normalize = transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
     data_transforms = {
-       	'swap': transforms.Compose([
-            transforms.Randomswap((swap_num[0], swap_num[1])),
-        ]),
+
         'common_aug': transforms.Compose([
             transforms.Resize((resize_reso, resize_reso)),
             transforms.RandomRotation(degrees=15), # 将图片进行随机旋转
@@ -35,8 +33,7 @@ def load_data_transformers(resize_reso=512, crop_reso=448, swap_num=[7, 7]):
             transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
         ]),
         'test_totensor': transforms.Compose([
-            transforms.Resize((resize_reso, resize_reso)),
-            transforms.CenterCrop((crop_reso, crop_reso)),
+            transforms.Resizepadding((crop_reso, crop_reso)),
             transforms.ToTensor(),
             transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
         ]),