aug_cartesian_product.py

import os
import numpy as np
from expmt_vars import dataset, tta_policy
import pdb
pct_opts = np.arange(.05, 1, .05)
AUG_ORDER = ['five_crop', 'hflip', 'colorjitter', 'rotation']
AUG_CART_PRODUCT = np.array([[1, 0, 0, 0], [1, 0, 0, 2], [1, 0, 0, 3], [1, 0, 1, 0], [1, 0, 1, 2], [1, 0, 1, 3], [1, 1, 0, 0], [1, 1, 0, 2], [1, 1, 0, 3], [1, 1, 1, 0], [1, 1, 1, 2], [1, 1, 1, 3], [3, 0, 0, 0], [3, 0, 0, 2], [3, 0, 0, 3], [3, 0, 1, 0], [3, 0, 1, 2], [3, 0, 1, 3], [3, 1, 0, 0], [3, 1, 0, 2], [3, 1, 0, 3], [3, 1, 1, 0], [3, 1, 1, 2], [3, 1, 1, 3], [4, 0, 0, 0], [4, 0, 0, 2], [4, 0, 0, 3], [4, 0, 1, 0], [4, 0, 1, 2], [4, 0, 1, 3], [4, 1, 0, 0], [4, 1, 0, 2], [4, 1, 0, 3], [4, 1, 1, 0], [4, 1, 1, 2], [4, 1, 1, 3], [2, 0, 0, 0], [2, 0, 0, 2], [2, 0, 0, 3], [2, 0, 1, 0], [2, 0, 1, 2], [2, 0, 1, 3], [2, 1, 0, 0], [2, 1, 0, 2], [2, 1, 0, 3], [2, 1, 1, 0], [2, 1, 1, 2], [2, 1, 1, 3], [0, 0, 0, 0], [0, 0, 0, 2], [0, 0, 0, 3], [0, 0, 1, 0], [0, 0, 1, 2], [0, 0, 1, 3], [0, 1, 0, 0], [0, 1, 0, 2], [0, 1, 0, 3], [0, 1, 1, 0], [0, 1, 1, 2], [0, 1, 1, 3]])
if os.path.exists('./' + dataset + '/' + tta_policy + '/aug_list.npy'):
    AUG_ORDER = np.load('./'+ dataset + '/' + tta_policy  + '/aug_order.npy')
    AUG_CART_PRODUCT = np.load('./'+ dataset + '/' + tta_policy +  '/aug_list.npy')