00001 import sys 00002 00003 00004 # class config: 00005 # basepath='/home/aramisa/datasets/PoloDataset/' 00006 # num_vwords=[1024] 00007 # test_list='lists/test50_blau.txt' 00008 # det_prob=0.4 00009 # min_overlap=0.5 00010 # gt_file='gt/necks_iripolo.txt' 00011 # merge_windows=False 00012 # verb=True 00013 # refine_nonlinear=False 00014 00015 class config: 00016 width=640 00017 height=480 00018 basepath='/home/dmartinez/experimentos/polo_live/' #'/home/aramisa/datasets/PoloDataset12/' 00019 num_vwords=[1024] 00020 det_prob=0.4 00021 min_overlap=0.5 00022 train_list='gt/necks_iripolo.txt' 00023 gt_file='gt/necks_iripolo.txt' 00024 merge_windows=False 00025 verb=2 00026 refine_nonlinear=False 00027 #neg_windows='full' 00028 neg_windows='random' 00029 single_pred=False 00030 use_mask=False 00031 00032 class polo_live(config): 00033 def __init__(self): 00034 k_gih='64' 00035 k_sift='128' 00036 self.vectors_suffix='_gih_features_k'+k_gih+'_lava_dense_sift_'+k_sift 00037 self.words_path=['tmp/gih/', 'tmp/sift/'] 00038 self.classifier_suffix='_gih_features_'+k_gih+'_lava_dense_sift_'+k_sift 00039 self.det_results_dir='detect_v3_results_gih_features_k'+k_gih+'_lava_dense_sift_k'+k_sift+'/' 00040 self.num_vwords=[int(k_gih),int(k_sift)] 00041 self.merge_windows=True 00042 #self.min_overlap=0.4 00043 self.refine_nonlinear=True 00044 00045 class polo_live_sift(config): 00046 def __init__(self): 00047 k_sift='128' 00048 self.vectors_suffix='_sift_k'+k_sift 00049 self.words_path=['tmp/sift/'] 00050 self.classifier_suffix='_dense_sift_'+k_sift 00051 self.det_results_dir='detect_v3_results_lava_dense_sift_k'+k_sift+'/' 00052 self.num_vwords=[int(k_sift)] 00053 self.merge_windows=True 00054 #self.min_overlap=0.4 00055 self.refine_nonlinear=True 00056 self.use_mask=True 00057 00058 if len(sys.argv)>1: 00059 if '(' in sys.argv[1]: 00060 func=sys.argv[1].split('(')[0] 00061 args=sys.argv[1].split('(')[1].split(')')[0] 00062 if ',' in args: 00063 args=args.split(',') 00064 else: 00065 args=[args] 00066 print args 00067 00068 object = globals()[func] 00069 conf=object(*args) 00070 else: 00071 object = globals()[sys.argv[1]] 00072 conf=object() 00073 else: 00074 #conf=config_fpfh() 00075 #conf=config_sift() 00076 #conf=config_fpfh_sift() 00077 #conf=config_fpfh_sift_v3_test_others() 00078 #conf=config_sift_v3() 00079 #conf=config_fpfh_v3() 00080 #conf=config_fpfh_v3_test_others() 00081 #conf=config_hks_sparse() 00082 #conf=config_hks_sparse_v3() 00083 #conf=config_sift_hks_sparse() 00084 #conf=config_sift_hks_sparse_v3() 00085 conf=config_hks_basic_time() 00086 #conf=config_hks_basic_time_v3()