From 5c53274c677773dae82ac66a9ddd4770ebbbe0c6 Mon Sep 17 00:00:00 2001
From: yayoimizuha <poko.peko.tomo@icloud.com>
Date: Thu, 20 Apr 2023 01:27:04 +0900
Subject: [PATCH] update

---
 resnet_finetune.py  | 4 +++-
 torch_dataloader.py | 4 ++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/resnet_finetune.py b/resnet_finetune.py
index 8437306..df29c91 100644
--- a/resnet_finetune.py
+++ b/resnet_finetune.py
@@ -3,7 +3,7 @@ from torchvision.models import ResNet50_Weights, resnet50
 from torch.nn import Linear
 from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, ToTensor, \
     RandomHorizontalFlip, \
-    Resize, CenterCrop, RandomAffine
+    Resize, CenterCrop, RandomAffine, GaussianBlur, RandomAutocontrast
 import matplotlib.pyplot as plt
 from numpy import arange
 from torchsummary import summary
@@ -23,6 +23,8 @@ transform = {
     'train': Compose([
         CenterCrop(200),
         RandomHorizontalFlip(p=0.1),
+        GaussianBlur(kernel_size=3),
+        RandomAutocontrast(),
         # Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
         ToTensor(),
         RandomRotation(degrees=15),
diff --git a/torch_dataloader.py b/torch_dataloader.py
index bc65db4..db25df8 100644
--- a/torch_dataloader.py
+++ b/torch_dataloader.py
@@ -18,6 +18,6 @@ def show_image(x):
         plt.imshow(img)
 
 
-model = models.inception_v3(pretrained=True)
+model = models.inception_v3(models.Inception_V3_Weights)
 
-print(model)
+print(model._get_name())