swirlspy.qpf.dl.helpers package

Submodules

swirlspy.qpf.dl.helpers.gifmaker module

swirlspy.qpf.dl.helpers.gifmaker.save_gif(single_seq, fname)

Save a single gif consisting of image sequence in single_seq to fname.

swirlspy.qpf.dl.helpers.gifmaker.save_gifs(seq, prefix)

Save several gifs.

Parameters:
  • seq (Shape (num_gifs, IMG_SIZE, IMG_SIZE)) –
  • prefix (prefix-idx.gif will be the final filename.) –

swirlspy.qpf.dl.helpers.log_analysis module

swirlspy.qpf.dl.helpers.log_analysis.parse_log(file_path, regex)
Parameters:
  • file_path –
  • regex –
swirlspy.qpf.dl.helpers.log_analysis.remove_duplicates_and_convert_npy(val_list)
swirlspy.qpf.dl.helpers.log_analysis.temporal_smoothing(training_statistics, stride=10, window_size=100)

We always assume the first axis in statistics is the iteration

Parameters:
  • training_statistics –
  • stride –
  • window_size –
Returns:

  • smoothed_mean
  • smoothed_std

swirlspy.qpf.dl.helpers.msssim module

Python implementation of MS-SSIM. Usage: python msssim.py –original_image=original.png –compared_image=distorted.png

swirlspy.qpf.dl.helpers.msssim.MultiScaleSSIM(img1, img2, max_val=255, filter_size=11, filter_sigma=1.5, k1=0.01, k2=0.03, weights=None)

Return the MS-SSIM score between img1 and img2. This function implements Multi-Scale Structural Similarity (MS-SSIM) Image Quality Assessment according to Zhou Wang’s paper, “Multi-scale structural similarity for image quality assessment” (2003). Link: https://ece.uwaterloo.ca/~z70wang/publications/msssim.pdf Author’s MATLAB implementation: http://www.cns.nyu.edu/~lcv/ssim/msssim.zip

Parameters:
  • img1 (Numpy array holding the first RGB image batch.) –
  • img2 (Numpy array holding the second RGB image batch.) –
  • max_val (the dynamic range of the images (i.e., the difference between the) – maximum the and minimum allowed values).
  • filter_size (Size of blur kernel to use (will be reduced for small images)) –
  • filter_sigma (Standard deviation for Gaussian blur kernel (will be reduced) – for small images).
  • k1 (Constant used to maintain stability in the SSIM calculation (0.01 in) – the original paper).
  • k2 (Constant used to maintain stability in the SSIM calculation (0.03 in) – the original paper).
  • weights (List of weights for each level; if none, use five levels and the) – weights from the original paper.
Returns:
Return type:

MS-SSIM score between img1 and img2.
Raises:

RuntimeError: If input images don’t have the same shape or don’t have four – dimensions: [batch_size, height, width, depth].

swirlspy.qpf.dl.helpers.ordered_easydict module

class swirlspy.qpf.dl.helpers.ordered_easydict.OrderedEasyDict(d=None, **kwargs)

Bases: collections.OrderedDict

Using OrderedDict for the easydict package See Also https://pypi.python.org/pypi/easydict/

swirlspy.qpf.dl.helpers.visualization module

swirlspy.qpf.dl.helpers.visualization.flow_to_img(flow_dat, max_displacement=None)

Convert optical flow data to HSV images

Parameters:
  • flow_dat (np.ndarray) – Shape: (seq_len, 2, H, W)
  • max_displacement (float or None) –
Returns:

rgb_dat – Shape: (seq_len, 3, H, W)
Return type:

np.ndarray
swirlspy.qpf.dl.helpers.visualization.get_color_flow_legend_image(size=50)
swirlspy.qpf.dl.helpers.visualization.merge_rgba_cv2(front_img, back_img)

Merge the front image with the background image using the Painter’s algorithm

Parameters:
  • front_img (np.ndarray) –
  • back_img (np.ndarray) –
Returns:

result_img
Return type:

np.ndarray
swirlspy.qpf.dl.helpers.visualization.save_hko_gif(im_dat, save_path)

Save the HKO images to gif

Parameters:
  • im_dat (np.ndarray) – Shape: (seqlen, H, W)
  • save_path (str) –
swirlspy.qpf.dl.helpers.visualization.save_hko_movie(im_dat, datetime_list, mask_dat=None, save_path='hko.mp4', masked=False, fps=5, prediction_start=None)

Save the HKO images to a video file

Parameters:
  • im_dat (np.ndarray) – Shape : (seq_len, H, W)
  • datetime_list (list) – list of datetimes
  • mask_dat (np.ndarray or None) – Shape : (seq_len, H, W)
  • save_path (str) –
  • masked (bool) – whether the mask the inputs when saving the image
  • fps (float) – the fps of the saved movie
  • prediction_start (int or None) – The starting point of the prediction