0
点赞
收藏
分享

微信扫一扫

paddlepaddle 1 数据加载器的使用

1、生成训练txt

import os
import random
from random import shuffle

def save_train_line(path,save_name,ext='.png',mode='w'):
    temp_seg = os.listdir(path)
    ftrainval = open(save_name, mode)
    for name in temp_seg:
        if name.endswith(ext):
            ftrainval.write(path + "\\" + name + '\n')
            print(path + "\\" + name + '\n')
    ftrainval.close()
def save_train_val(path,train_name,val_name,mode='w',train_rate=0.7,ext='.png'):
    temp_seg = os.listdir(path)
    shuffle(temp_seg)
    train_len=int(train_rate*len(temp_seg))
    ftrainval = open(train_name, mode)
    for i in range(train_len):
        name=temp_seg[i]
        if name.endswith(ext):
            ftrainval.write(path + "\\" + name + '\n')
            print(path + "\\" + name + '\n&
举报

相关推荐

0 条评论