def write_thumb(inferer, state, data_dir, store_name):
store_path = os.path.join(data_dir, store_name)
thumb = inferer.prepare_thumb(state)
imageio.imwrite(store_path, thumb)
微信
公众号
扫码关注公众号