pull in dulldream celery server

1 files changed, 20 insertions, 0 deletions

diff --git a/server/app/main/img_proc_config.py b/server/app/main/img_proc_config.py
new file mode 100644
index 00000000..db124978
--- /dev/null
+++ b/server/app/main/img_proc_config.py
@@ -0,0 +1,20 @@
+# 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'