python生成训练集和测试集,原数据以txt给出

源数据: 第二列为id号; 将以txt存储的图片分成train和test集,train和test比例按每个id的总图片数的7:3,实现代码如下所示: id_a = [] #存放所有的标签号 id_f = [] #存放图片数大于等于20的的标签号 id_num =[] path = '' save_train_path = '' save_val_path = '' with codecs.open
相关文章
相关标签/搜索