Maternetworkin87 commited on
Commit
253c98d
1 Parent(s): 7da862c

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -2
model.py CHANGED
@@ -36,8 +36,8 @@ def create_effnetb2_model(num_classes:int=100,
36
 
37
  return model
38
 
39
- def manual_transforms(img)
40
- transforms.Compose([
41
  transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes)
42
  transforms.ToTensor(), # 2. Turn image values to between 0 & 1
43
  transforms.Normalize(mean=[0.485, 0.456, 0.406], # 3. A mean of [0.485, 0.456, 0.406] (across each colour channel)
 
36
 
37
  return model
38
 
39
+ def manual_transforms(img):
40
+ img = transforms.Compose([
41
  transforms.Resize((224, 224)), # 1. Reshape all images to 224x224 (though some models may require different sizes)
42
  transforms.ToTensor(), # 2. Turn image values to between 0 & 1
43
  transforms.Normalize(mean=[0.485, 0.456, 0.406], # 3. A mean of [0.485, 0.456, 0.406] (across each colour channel)