duhaime

import warnings
from skimage.measure import compare_ssim
from skimage.transform import resize
from scipy.stats import wasserstein_distance
from scipy.misc import imsave
from scipy.ndimage import imread
import numpy as np
import cv2

##

Tushar-N

import torch
import torch.nn as nn
import torch.nn.functional as F

import torchvision.models as tmodels
from functools import partial
import collections

# dummy data: 10 batches of images with batch size 16
dataset = [torch.rand(16,3,224,224).cuda() for _ in range(10)]

Tushar-N

import cv2
import numpy as np
import argparse

'''
usage: python click_heatmap.py <image file>
    left-click: add point to heatmap
    s: save image (000.png, 001.png, ...) 
    q: quit
    r: reset