【教学】python基于vgg的CT_COVID与CT_NonCOVID二分类识别+pyqt界面含代码

效果演示图如下:

代码按照顺序运行即可

 vgg训练模型的代码如下:

import torch
from PIL import Image
from torchvision import datasets, models, transforms,utils
import torch.nn as nn
import numpy as np
import random
import os
import torchvision
from tqdm import tqdm

#设置随机种子

def setup_seed(seed):
   torch.manual_seed(seed)
   torch.cuda.manual_seed(seed)
   torch.cuda.manual_seed_all(seed)
   np.random.seed(seed)
   random.seed(seed)
   torch.backends.cudnn.benchmark = False
   torch.backends.cudnn.deterministic = True
   torch.backends.cudnn.enabled = True
   os.environ['PYTHONHASHSEED'] = str(seed)
setup_seed(20)

root = './'

# Hyper parameters
num_epochs = 3    #循环次数
batch_size 

猜你喜欢

转载自blog.csdn.net/qq_34904125/article/details/124918816