def save_samples(samples, img_path, vis_fn):
batch_size = len(samples)
rows, cols = utils.factorize_number(batch_size)
vis_fn(samples, rows, cols, img_path)
微信
公众号
扫码关注公众号