def test_create_mean2(self): """ calc_mean case2: 2 dataset """ filepath = 'resources/test_train/image-list2.txt' meanpath = 'resources/test_train/mean.npy' if os.path.exists(meanpath): os.remove(meanpath) dataset.create_mean(filepath, meanpath) assert os.path.exists(meanpath)
def genmean(args): dataset.create_mean(args.train, args.output)