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

update

This commit is contained in:
yayoimizuha 2023-04-16 11:38:44 +09:00
parent 2e2ce43e30
commit fa51c24f46
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
inceptionnet_finetune.py
View File

@ -1,5 +1,6 @@
from os import makedirs from os import makedirs
from torchvision.models import Inception_V3_Weights, inception_v3 from torchvision.models import Inception_V3_Weights, inception_v3
from torchvision.models import swin_v2_b, Swin_V2_B_Weights
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, \