效果图如下:
代码主要要包含两个部分。
一个是算法部分,一个是小程序部分。通过上面途径可以下载到完整的代码,包括说明文档,如何安装环境运行代码。
算法部分主要运行训练demo如下:
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)
r