tomokazu
/
helloproject-ai
mirror of https://github.com/yayoimizuha/helloproject-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update
continuous-integration/drone/push Build is passing Details

This commit is contained in:
yayoimizuha 2023-04-20 01:27:04 +09:00
parent e4308ca509
commit 5c53274c67
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
resnet_finetune.py
View File

@ -3,7 +3,7 @@ from torchvision.models import ResNet50_Weights, resnet50
from torch.nn import Linear from torch.nn import Linear
from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, ToTensor, \ from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, ToTensor, \
RandomHorizontalFlip, \ RandomHorizontalFlip, \
Resize, CenterCrop, RandomAffine Resize, CenterCrop, RandomAffine, GaussianBlur, RandomAutocontrast
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from numpy import arange from numpy import arange
from torchsummary import summary from torchsummary import summary
@ -23,6 +23,8 @@ transform = {
'train': Compose([ 'train': Compose([
CenterCrop(200), CenterCrop(200),
RandomHorizontalFlip(p=0.1), RandomHorizontalFlip(p=0.1),
GaussianBlur(kernel_size=3),
RandomAutocontrast(),
# Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), # Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
ToTensor(), ToTensor(),
RandomRotation(degrees=15), RandomRotation(degrees=15),

4
torch_dataloader.py
View File

@ -18,6 +18,6 @@ def show_image(x):
plt.imshow(img) plt.imshow(img)
model = models.inception_v3(pretrained=True) model = models.inception_v3(models.Inception_V3_Weights)
print(model) print(model._get_name())