blob: db124978a1ea80455ac04653c10b11d6c4243e64 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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'
|
