def startup(self, args={}): print('无锡所应用...') MMongoDb._initialize() run_mode = 1 if 1 == run_mode: #WxsDsm.generate_txt_by_wxs_tds_ok_images() WxsDsm.generate_ds_from_wxs_tds_ibc() elif 2 == run_mode: WxsDsm.generate_wxs_tds_for_integration_test()
def get_bmy_code_to_bmy_id_dict(): ''' 获取bmy_code到bmy_id字典 ''' query_cond = {} fields = {'bmy_code': 1, 'bmy_id': 1} return MMongoDb.convert_recs(MMongoDb.db['t_bmy']\ .find(query_cond, fields))
def get_bmy_id_model_ids(): ''' 获取bmy_id对应model_id的对应关系 ''' query_cond = {} fields = {'bmy_id': 1, 'model_id': 1} return MMongoDb.convert_recs(MMongoDb.db['t_bmy']\ .find(query_cond, fields))
def is_sample_exists(img_file): query_cond = {'img_file': img_file} fields = {'sample_id': 1, 'bmy_id': 1} rec = MMongoDb.convert_rec(MMongoDb.db['t_sample'].find_one( query_cond, fields)) if rec is None: return False else: return True
def get_bmys(): query_cond = {} fields = { 'bmy_id': 1, 'bmy_name': 1, 'bmy_code': 1, 'brand_id': 1, 'brand_code': 1, 'model_id': 1, 'model_code': 1 } return MMongoDb.convert_recs(MMongoDb.db['t_bmy']\ .find(query_cond, fields).sort([('bmy_id', pymongo.ASCENDING)]))
def get_bmy_id_bmy_vos(): query_cond = {} fields = { 'bmy_id': 1, 'bmy_name': 1, 'bmy_code': 1, 'brand_id': 1, 'brand_code': 1, 'model_id': 1, 'model_code': 1 } return MMongoDb.convert_recs(MMongoDb.db['t_bmy']\ .find(query_cond, fields))
def get_bmy_id_bmy_names(): query_cond = {} fields = {'bmy_id': 1, 'bmy_name': 1} return MMongoDb.convert_recs(MMongoDb.db['t_bmy'].find( query_cond, fields))
def get_wxs_brands(): query_cond = {'source_type': 1} fields = {'brand_id': 1, 'brand_name': 1, 'brand_code': 1} return MMongoDb.convert_recs(MMongoDb.db['t_brand']\ .find(query_cond, fields))
def get_vin_samples(vin_id): query_cond = {'vin_id': vin_id} fields = {"img_file": 1, "bmy_id": 1} return MMongoDb.convert_recs(MMongoDb.db['t_sample'].find( query_cond, fields).sort([('img_file', 1)]))
def get_vins_base(source_type): query_cond = {'source_type': source_type} fields = {'vin_id': 1, 'vin_code': 1, 'bmy_id': 1} return MMongoDb.convert_recs(MMongoDb.db['t_vin']\ .find(query_cond, fields).sort([('vin_code', 1)]))
def get_vin_code_bmy_id_dict(): query_cond = {} fields = {'vin_code': 1, 'bmy_id': 1} return MMongoDb.convert_recs(MMongoDb.db['t_vin']\ .find(query_cond, fields))
def get_vin_codes(): query_cond = {} fields = {"vin_id": 1, "vin_code": 1} return MMongoDb.convert_recs(MMongoDb.db['t_vin'].find( query_cond, fields).sort([('vin_code', 1)]))
def get_bmy_ids_by_vin_code(prefix_vin_code): query_cond = {"vin_code": prefix_vin_code} fields = {"vin_id": 1, "bmy_id": 1} return MMongoDb.convert_recs(MMongoDb.db['t_vin'].find( query_cond, fields))
def get_bmy_id_by_vin_code(vin_code): query_cond = {"vin_code": vin_code} fields = {"vin_id": 1, "bmy_id": 1} return MMongoDb.convert_rec(MMongoDb.db['t_vin'].find_one( query_cond, fields))
def main(args): MMongoDb._initialize() print('细粒度图像识别系统') mode = MODE_WXS_BID if MODE_DRAW_ACCS_CURVE == mode: #du.draw_accs_curve() pass elif MODE_WXS_BID == mode: app = WxsApp() app.startup(args) elif MODE_DRAW_CAM == mode: app = CamApp() app.startup(args) elif MODE_SIAMESE_NETWORK == mode: app = SiameseApp() app.startup(args) elif MODE_TRAIN_WEB_SERVER == mode: print('训练过程...') elif MODE_RUN_WEB_SERVER == mode: ''' web_server = WebServer() web_server.setDaemon(True) web_server.start() web_server.join() ''' pass elif MODE_TRAIN_MONITOR == mode: ''' web_server = WebServer() web_server.setDaemon(True) web_server.start() # start training process for i in range(100): try: time.sleep(3) except: print('exceptions... stop thread') break print('loop_{0}: lr={1};'.format(i, app_store.vars['lr'])) ''' pass elif MODE_GET_BEST_CHPTS == mode: get_best_chpts() elif MODE_CREATE_ST_CAR_DS == mode: #du.prepare_st_car_ds() pass elif MODE_VAO_TEST == mode: VaoTest.startup() elif MODE_DATA_PREPROCESSOR == mode: DataPreprocessor.startup() elif MODE_DS_MANAGER == mode: DsManager.startup() elif MODE_CLUSTER_IMAGE == mode: app = ClusterApp() app.startup() elif MODE_LOCAL_STANFORD_CARS == mode: DataPreprocessor.startup() elif MODE_TEST_MONGODB == mode: app = VbgApp() app.startup() elif MODE_TEST_ADMIN == mode: AdminApp.startup() elif MODE_TEST_WEB_API == mode: test_web_api() else: print('临时测试程序...') temp_func()
def get_dataset_sample_by_infos(dataset_id, sample_id, sample_type): tbl = MMongoDb.db['t_dataset_sample'] query_cond = {'dataset_id': dataset_id, 'sample_id': sample_id, 'sample_type': sample_type} fields = {'dataset_sample_id': 1, 'dataset_id': 1, 'sample_id': 1, 'sample_type': 1} return MMongoDb.convert_rec(tbl.find_one(query_cond, fields))
def get_wxs_bms(): query_cond = {'source_type': 1} fields = {'model_code': 1, 'model_name': 1} return MMongoDb.convert_recs(MMongoDb.db['t_model']\ .find(query_cond, fields))
def get_vin_id_codes(): query_cond = {'vin_code': {'$regex': '_'}} fields = {'vin_id': 1, 'vin_code': 1, 'bmy_id': 1} return MMongoDb.convert_recs(MMongoDb.db['t_vin']\ .find(query_cond, fields).sort([('vin_code', 1)]))