| |
| import numpy as np |
| from NER.poi_processing import filter_poi_list |
|
|
| def find_nearest(array, a): |
| array = np.array(array) |
| if type(a) in [float,int] or len(a)==1: |
| distances = np.abs(array - a) |
| else: |
| distances = np.linalg.norm(array - a, axis=1) |
| min_index = np.argmin(distances) |
| return min_index |
|
|
| def sec2gps(video_csv,sec=0): |
| |
| idx = find_nearest(video_csv['video_tps'],sec) |
| return video_csv.iloc[idx]['latitude'],video_csv.iloc[idx]['longitude'] |
|
|
|
|
| def get_video_location_and_poi_data(task, xinyi_csv, sec17_csv, xinyi_poi_data, zhongxiao_poi_data): |
| """ |
| 根據 task 決定使用的 video_location_data 和 poi_data |
| """ |
| |
| if task == 'xinyi': |
| video_location_data = xinyi_csv |
| poi_data = xinyi_poi_data |
| elif task == 'zhongxiao': |
| video_location_data = sec17_csv |
| poi_data = zhongxiao_poi_data |
| |
| poi_data = filter_poi_list(poi_data) |
| return video_location_data, poi_data |
|
|
|
|
| def get_ner_coordinates(task, timestamp, xinyi_csv, sec17_csv): |
| """ |
| 計算 NER 座標位置並將 timestamp 減去 1 秒 |
| """ |
| |
| if task == 'xinyi': |
| timestamp = min(max(5, timestamp - 1), 1494) |
| ner_cur_pos = sec2gps(xinyi_csv, timestamp) |
| elif task == 'zhongxiao': |
| timestamp = min(max(5, timestamp - 1), 2514) |
| ner_cur_pos = sec2gps(sec17_csv, timestamp) |
| return ner_cur_pos |
|
|
| import re |
| def remove_chinese_characters(text): |
| tmp = re.sub(r'[\u4E00-\u9FFF\u3400-\u4DBF]+','', text) |
| return tmp |
| return re.sub(r'\[.*?\]', '', tmp) |
