代码拉取完成,页面将自动刷新
import os
import glob
import pickle
def get_image_paths(images_dir, ext):
return glob.glob(os.path.join(images_dir, "*" + ext))
def get_filename(image_path):
filename_w_ext = os.path.basename(image_path)
filename = os.path.splitext(filename_w_ext)[0]
return filename
def get_file_ext(image_path):
filename_w_ext = os.path.basename(image_path)
ext = os.path.splitext(filename_w_ext)[1]
return ext
def save_params(exp_folder, **params):
file_name = 'params.pkl'
file_path = os.path.join(exp_folder, file_name)
with open(file_path, 'wb') as f:
pickle.dump(params, f, pickle.HIGHEST_PROTOCOL)
def load_params(exp_folder):
file_name = 'params.pkl'
file_path = os.path.join(exp_folder, file_name)
with open(file_path, 'rb') as f:
return pickle.load(f)
def num_iter_per_epoch(num_data, batch_size):
return int(num_data // batch_size + (num_data % batch_size > 0))
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。