当前位置:   article > 正文

ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be emp_the resulting train set will be empty

the resulting train set will be empty

今天写代码labelmetovoc,即将labelme标注的转化为voc标准格式参考的这篇文章时遇到了如下问题:
ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.
在网上查了一下,大部分博客都认为是scikit-learn版本较高时出现的问题,需要换到0.20.0以下版本,但是换的话numpy,scipy等库均要重新下载兼容的版本,所以建议不要轻易尝试,我请教了一下师兄,他说是因为参数上面参数设置出现了问题,train或test的数量必须为整数,改好的代码如下所示:

# -*- coding: utf-8 -*-
"""
Created on Fri Apr 10 13:39:17 2020

@author: nihao
"""
import os
import numpy as np
import codecs
import json
import glob
import cv2
import shutil
from sklearn.model_selection import train_test_split

# 1.标签路径
labelme_path = "D:\\PinInspection\\jpgretanglelabel"  # 原始labelme标注数据路径
saved_path = "D:\\PinInspection\\improveto300\\VOC2007"  # 保存路径

# 2.创建要求文件夹
dst_annotation_dir = os.path.join(saved_path, 'Annotations')
if not os.path.exists(dst_annotation_dir):
    os.makedirs(dst_annotation_dir)
dst_image_dir = os.path.join(saved_path, "JPEGImages")
if not os.path.exists(dst_image_dir):
    os.makedirs(dst_image_dir)
dst_main_dir = os.path.join(saved_path, "ImageSets", "Main")
if not os.path.exi
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/726866
推荐阅读
  

闽ICP备14008679号