【教学】基于mobilenet的水果分类识别小程序

效果图如下:

代码主要要包含两个部分。

一个是算法部分,一个是小程序部分。通过上面途径可以下载到完整的代码,包括说明文档,如何安装环境运行代码。

 算法部分主要运行训练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

猜你喜欢

转载自blog.csdn.net/qq_34904125/article/details/124854899
今日推荐