|
@@ -253,7 +253,7 @@ def predict():
|
|
|
# 上传redis
|
|
|
key_name = config_.RECALL_KEY_NAME_PREFIX + time.strftime('%Y%m%d')
|
|
|
redis_helper = RedisHelper()
|
|
|
- redis_helper.add_data_with_zset(key_name=key_name, data=applet_redis_data)
|
|
|
+ redis_helper.add_data_with_zset(key_name=key_name, data=applet_redis_data, expire_time=2*24*3600)
|
|
|
log_.info('applet data to redis finished!')
|
|
|
|
|
|
# 清空修改ROV的视频数据
|
|
@@ -290,7 +290,7 @@ def predict():
|
|
|
# 上传Redis
|
|
|
redis_helper = RedisHelper()
|
|
|
app_key_name = config_.RECALL_KEY_NAME_PREFIX_APP + time.strftime('%Y%m%d')
|
|
|
- redis_helper.add_data_with_zset(key_name=app_key_name, data=app_redis_data)
|
|
|
+ redis_helper.add_data_with_zset(key_name=app_key_name, data=app_redis_data, expire_time=2*24*3600)
|
|
|
log_.info('app data to redis finished!')
|
|
|
# 清空修改ROV的视频数据
|
|
|
redis_helper.del_keys(key_name=config_.UPDATE_ROV_KEY_NAME_APP)
|
|
@@ -323,7 +323,7 @@ def predict_test():
|
|
|
# 上传Redis
|
|
|
redis_helper = RedisHelper()
|
|
|
key_name = config_.RECALL_KEY_NAME_PREFIX + time.strftime('%Y%m%d')
|
|
|
- redis_helper.add_data_with_zset(key_name=key_name, data=redis_data)
|
|
|
+ redis_helper.add_data_with_zset(key_name=key_name, data=redis_data, expire_time=2*24*3600)
|
|
|
log_.info('test data to redis finished!')
|
|
|
# 清空修改ROV的视频数据
|
|
|
redis_helper.del_keys(key_name=config_.UPDATE_ROV_KEY_NAME)
|
|
@@ -356,7 +356,7 @@ def predict_test():
|
|
|
# 上传Redis
|
|
|
redis_helper = RedisHelper()
|
|
|
app_key_name = config_.RECALL_KEY_NAME_PREFIX_APP + time.strftime('%Y%m%d')
|
|
|
- redis_helper.add_data_with_zset(key_name=app_key_name, data=app_redis_data)
|
|
|
+ redis_helper.add_data_with_zset(key_name=app_key_name, data=app_redis_data, expire_time=2*24*3600)
|
|
|
log_.info('app test data to redis finished!')
|
|
|
# 清空修改ROV的视频数据
|
|
|
redis_helper.del_keys(key_name=config_.UPDATE_ROV_KEY_NAME_APP)
|
|
@@ -441,14 +441,14 @@ def predict_18_19():
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
- # log_.info('rov model train start...')
|
|
|
- # train_start = time.time()
|
|
|
- # train_filename = config_.TRAIN_DATA_FILENAME
|
|
|
- # X, Y, videos, fea = process_data(filename=train_filename)
|
|
|
- # log_.info('X_shape = {}, Y_sahpe = {}'.format(X.shape, Y.shape))
|
|
|
- # train(X, Y, features=fea)
|
|
|
- # train_end = time.time()
|
|
|
- # log_.info('rov model train end, execute time = {}ms'.format((train_end - train_start)*1000))
|
|
|
+ log_.info('rov model train start...')
|
|
|
+ train_start = time.time()
|
|
|
+ train_filename = config_.TRAIN_DATA_FILENAME
|
|
|
+ X, Y, videos, fea = process_data(filename=train_filename)
|
|
|
+ log_.info('X_shape = {}, Y_sahpe = {}'.format(X.shape, Y.shape))
|
|
|
+ train(X, Y, features=fea)
|
|
|
+ train_end = time.time()
|
|
|
+ log_.info('rov model train end, execute time = {}ms'.format((train_end - train_start)*1000))
|
|
|
|
|
|
log_.info('rov model predict start...')
|
|
|
predict_start = time.time()
|