diff options
Diffstat (limited to 'old/server/app/main/img_proc_config.py')
| -rw-r--r-- | old/server/app/main/img_proc_config.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/old/server/app/main/img_proc_config.py b/old/server/app/main/img_proc_config.py new file mode 100644 index 00000000..db124978 --- /dev/null +++ b/old/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' |
