#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
This started as a copy of https://bitbucket.org/RSKothari/multiset_gaze/src/master/ 
with additional changes and modifications to adjust it to our implementation. 

Copyright (c) 2021 Rakshit Kothari, Aayush Chaudhary, Reynold Bailey, Jeff Pelz, 
and Gabriel Diaz
"""
from models.resnet.res_18_eye import res_18_eye
from models.resnet.res_18_gaze import res_18_gaze
from models.resnet.res_18_gpm import res_18_gpm

# from models.DenseElNet_old import DenseNet2D

model_dict={'res_18_gaze': res_18_gaze,
            'res_18_gpm': res_18_gpm,
            'res_18_eye': res_18_eye,
            }