On_theway10的博客

paddle.fluid.io.DataLoaderfrom_generatorfrom_generator(feed_list=None, capacity=None, use_double_buffer=True, iterable=True, return_list=False, use_multiprocess=False, drop_last=Trueimport paddle.fluid as fluidimport numpy as npBATCH_NUM = 10BA

paddle.batch(reader, batch_sizedef make_reader(txt_file, is_train = True): def reader(): with open(txt_file, 'r') as f: lines = f.readlines() f.close() if is_train: np.random.shuffle(lines) .

freezemodel.inferenceclass Antispoof: def __init__(self, resume = '', use_gpu = True): self.use_gpu = use_gpu self.place = fluid.CUDAPlace(0) if self.use_gpu else fluid.CPUPlace() self.exe = fluid.Executor(sel.

model.freeze#-*- coding:utf-8 -*-import osimport paddle import paddle.fluid as fluidfrom model_zoos import antisp as modeldef gen_exe(resume = '', use_cuda = False): place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace() exe = fl.