# __init__.py
# Vojtěch Golda
from models.ssdlite import SSDlite
from models.ssd import SSD
from models.retina import RetinaNet
from models.r_cnn_mobilenet import RCNNMobileNet
from models.r_cnn_resnet import RCNNResNet
from models.fcos import FCOS
from models.tram_net_v1 import TramNetV1
from models.tram_net_v2 import TramNetV2

models = {
    "Faster_R_CNN_MobileNet": RCNNMobileNet,
    "FCOS": FCOS,
    "Faster_R_CNN_ResNet": RCNNResNet,
    "SSD": SSD,
    "SSDLite": SSDlite,
    "RetinaNet": RetinaNet,
    "TramNet" : TramNetV2,
}

# colors for multi network detections
# defined here to make it easy to see which is which 
colors = [
    (255,0,0),
    (0,255,0),
    (0,0,255),
    (0,255,255),
    (255,0,255),
    (255,255,0),
    (255,255,255),
]
