def load_csv(csv_name='../data/test.csv', need_save=0):
with open(csv_name, 'r') as f:
reader = csv.reader(f)
result_list = list(reader)
# result_list
# result_list = []
# for row in reader:
# result_list.append(row)
result = np.asarray(result_list)
# resultresult.astype('float16')
return result.astype('float32')
def load_list(video_list_name = 'list.txt'):
video_list = []
with open(video_list_name, 'r', encoding='utf-8') as f:
reader = csv.reader(f)
result_list = list(reader)
for video_name in result_list:
video_list.append(video_name[0].split('\t'))
return video_list