# paths for image processors
import os
from os.path import join

class ImgProcConfig:

	def __init__(self):
		dir_models = '/data_store/apps/dulldream/dnn_models'

		# mask rcnn
		self.mask_rcnn_class_config = '/dulldream/src/config/coco_meta.json'
		self.mask_rcnn_model = join(dir_models,'tf/mask_rcnn_coco.h5')

		# p2p
		self.p2p_ckpts_dir = join(dir_models,'p2p/coco2014_person')
		self.p2p_epoch = 'latest'

		# p2p objects only
		self.p2p_bg_ckpts_dir = join(dir_models,'p2p/coco2014_objects')
		self.p2p_bg_epoch = 'latest'