Start implementing Inception v3

This commit is contained in:
2024-05-19 17:43:42 +02:00
parent 2118cd47c3
commit d0b974dd9f
2 changed files with 65 additions and 0 deletions

View File

@@ -0,0 +1,17 @@
import torch
import torchvision
import sys
from torchsummary import summary
inception = torchvision.models.inception_v3(weights=torchvision.models.Inception_V3_Weights.DEFAULT)
inception.eval()
sys.path.append('../../tools') # Ugly hack
from utils import export_model_weights, print_model_parameters
print_model_parameters(inception) # print layer names and number of parameters
inception.cuda()
summary(inception, (3, 299, 299))