fix: performance (#38)

ResNet50 requires images with the size of 224, I had to increase that.

Also important changes are slight rotations which helps the model to
learn.

Increase the light of the images can help when images are a bit darker.

Those tiny changes increase the model accuracy from 71% to 80%

Author: GabrielLidenor

src/data_loader.py

@@ -11,6 +11,10 @@ def __init__(self, hf_dataset):
         self.data = hf_dataset
         self.transform = T.Compose([
             T.Grayscale(num_output_channels=3),
+            T.Resize((224,224)),
+            T.RandomHorizontalFlip(p=0.5),
+            T.RandomRotation(degrees=10),
+            T.RandomApply([T.ColorJitter(brightness=0.2, contrast=0.2)], p=0.5),
             T.ToTensor(),
             T.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
         ])