|
|
@@ -183,6 +183,7 @@ def predict():
|
|
|
predict_result = []
|
|
|
redis_data = {}
|
|
|
json_data = []
|
|
|
+ video_id_list = []
|
|
|
for j, item in enumerate(predict_data_sorted):
|
|
|
video_id = int(item['video_id'])
|
|
|
rov_score = 100 - j * config_.ROV_SCORE_D
|
|
|
@@ -190,6 +191,7 @@ def predict():
|
|
|
predict_result.append(item)
|
|
|
redis_data[video_id] = rov_score
|
|
|
json_data.append({'videoId': video_id, 'rovScore': rov_score})
|
|
|
+ video_id_list.append(video_id)
|
|
|
|
|
|
# 打包预测结果存入csv
|
|
|
predict_result_filename = 'predict.csv'
|
|
|
@@ -217,8 +219,8 @@ def predict():
|
|
|
log_.error('notify backend fail!')
|
|
|
|
|
|
# 更新视频的宽高比数据
|
|
|
- if video_ids:
|
|
|
- update_video_w_h_rate(video_ids=video_ids,
|
|
|
+ if video_id_list:
|
|
|
+ update_video_w_h_rate(video_ids=video_id_list,
|
|
|
key_name=config_.W_H_RATE_UP_1_VIDEO_LIST_KEY_NAME['rov_recall'])
|
|
|
log_.info('update video w_h_rate to redis finished!')
|
|
|
|
