import numpy as np
from utils.cityscape_utils import *

train_scenes = [
    "aachen",
    "bochum",
    "bremen",
    "cologne",
    "darmstadt",
    "dusseldorf",
    "erfurt",
    "hamburg",
    "hanover",
    "jena",
    "krefeld",
    "monchengladbach",
    "strasbourg",
    "stuttgart",
    "tubingen",
    "ulm",
    "weimar",
    "zurich"
]

train_scenes_extra = [
	"bochum",
	"hamburg",
	"hanover",
	"krefeld",
	"monchengladbach",
	"strasbourg"
]

val_scenes = [
    "frankfurt",	
    "lindau",		
    "munster"
]

val_scenes_extra = [
	"frankfurt"
]

test_scenes = [
    "berlin",  	 	
    "bielefeld", 	
    "bonn",			
    "leverkusen",	
    "mainz",		
    "munich"
]

test_scenes_extra = [
	"bielefeld",
	"mainz",

]


scene_dict = {
	'train': train_scenes,
	'val': val_scenes,
	'test': test_scenes
}

clip_num_dict = {
	'train':2975,
	'val': 500,
	'test': 1525
}

extra_scene_dict = {
	'train':train_scenes_extra,
	'val': val_scenes_extra,
	'test': test_scenes_extra
}

clip_id_start = {
	'train':16893,
	'val': 2097,
	'test': 9414
}



color_map = [
[128, 64, 128] ,   # road
[244, 35, 232]  ,  # sidewald
[70, 70, 70]	,		# building
[102, 102, 156]	,	# wall
[190, 153, 153]	,	# fence
[153, 153, 153]	,	# pole
[250, 170, 30]	,	# traffic light
[220, 220, 0]	,		# traffic sign
[107, 142, 35]	,	# vegetation
[152, 251, 152]	,	# terrain
[70, 130, 180]  ,	# sky
[220, 20, 60]	,	# person
[255, 0, 0]		,	# rider
[0, 0, 142]  	,	# car
[0, 0, 70]		,		# truck
[0, 60, 100]	,		# bus
[0, 80, 100]	,		# on rails / train
[0, 0, 230]		,	# motorcycle
[119, 11, 32]	,	# bicycle
[0, 0, 0]			# None
]
