实际应用时可能比较想获取VGG中间层的输出,那么就可以如下操作:
import numpy as np
import torch
from torchvision import models
from torch.autograd import Variable
import torchvision.transforms as transforms
class CNNShow():
def __init__(self, model):
self.model = model
self.model.