importcv2import numpy as npimport matplotlib.pyplot as pltimport matplotlib.imagefrom PIL.Image import Imagefrom numpy import ubytefrom torch import uint8o=cv2.imread("1.png",0)# 直方图均衡化处理（只能处理灰度图片）# c

import cv2from torch.utils.data import Datasetfrom PIL import Imageimport osimport matplotlib.pyplot as pltclass MyData(Dataset):# 初始化 root_dir大的路径和label_dir具体内容/获取地址def __init__(self, root_dir, label