1 Star 0 Fork 0

greitzmann/Keras-Image-Super-Resolution

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
utils.py 941 Bytes
一键复制 编辑 原始数据 按行查看 历史
hsbk 提交于 6年前 . full implementation
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))
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/greitzmann/Keras-Image-Super-Resolution.git
git@gitee.com:greitzmann/Keras-Image-Super-Resolution.git
greitzmann
Keras-Image-Super-Resolution
Keras-Image-Super-Resolution
master

搜索帮助