模型训练过后保存一直报错,提示不存在这个目录,但是这个目录是存在的,而且是可以读取和写入的

img

img


模型训练过后保存一直报错,提示不存在这个目录,但是这个目录是存在的,而且是可以读取和写入的

img


你这个给的是一个文件,他要的是一个目录

  • 这个问题的回答你可以参考下: https://ask.csdn.net/questions/7688713
  • 这篇博客你也可以参考下:训练自己的目标检测模型
  • 除此之外, 这篇博客: 手把手教你:铁路异物侵入的目标检测系统中的 1.1 加载图像数据,预处理,图像数据增强 部分也许能够解决你的问题, 你可以仔细阅读以下内容或跳转源博客中阅读:
  • #数据预处理
    #图片加载
    # 读取图片+数据处理函数
    def read_img(path):
        print("数据集地址:"+path)
        imgs = []
        labels = []
        for root, dirs, files in os.walk(path):
            for file in tqdm(files):
                # print(path+'/'+file+'/'+folder)
                # 读取的图片
                img = cv2.imread(os.path.join(root, file))
                # skimage.transform.resize(image, output_shape)改变图片的尺寸
                img = cv2.resize(img, (w, h))
                # 将读取的图片数据加载到imgs[]列表中
                imgs.append(img)
                # 将图片的label加载到labels[]中,与上方的imgs索引对应
                labels.append(str(os.path.basename(root)))
        return imgs,labels
    # 调用读取图片的函数,得到图片和labels的数据集
    data1, label1 = read_img(train_img_url)
    

    原始图片数量:
    1: 1090, 0: 572
    其中0为有异物的图片,1为无异物的正常图片。

    完成数据增强后数量:
    图像数量

  • 您还可以看一下 陈志文老师的项目经理的那些事课程中的 项目经理的薪资待遇以及发展前景小节, 巩固相关知识点