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)]))
