diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index b344ac686866a597a8662f322b3f07fc7ce2cc38..79ac4b3cfc1582e634c53c738b45c15755355763 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -4,7 +4,8 @@ import tensorflow as tf import util.util from util.setup import logdir, modeldir, cachepath, now, ancillary_path from util.util import EarlyStop, normalize, denormalize, resample, resample_2d_linear, resample_one,\ - resample_2d_linear_one, get_grid_values_all, add_noise, smooth_2d, smooth_2d_single + resample_2d_linear_one, get_grid_values_all, add_noise, smooth_2d, smooth_2d_single, median_filter_2d,\ + median_filter_2d_single, downscale_2x import os, datetime import numpy as np import pickle