Ejemplo n.º 1
0
def video_result(video_time, video_name):
    record_time_path = os.path.join(app.config['UPLOAD_FOLDER'], video_time)
    record_zip_folder_path = os.path.join(record_time_path, ZIP_FOLDER)
    record_list = os.listdir(record_zip_folder_path)
    for record in record_list:
        img_result_list = []
        record_folder = os.path.join(record_zip_folder_path, record)
        print video_name
        video_file = os.path.join(record_folder, video_name.replace('mp4', 'txt'))
        video_path = ''
        with open(video_file) as fh:
            lines = fh.readlines()
            for line in lines:
                img_result = ImageResult()
                line_arr = []
                line_arr = line.strip().split(' ')
                if line_arr.__len__() > 3:
                    pos = line_arr.__len__() - 1
                    img_result.score = line_arr[pos]
                    img_result.pre_label = line_arr[pos-1]
                    for i in range(pos - 1):
                        if i < pos - 2:
                            img_result.path += line_arr[i] + ' '
                        else:
                            img_result.path += line_arr[i]
                else:
                    img_result.path, img_result.pre_label, img_result.score = line.strip().split(' ')
                img_result.name = os.path.basename(img_result.path)
                img_result_list.append(img_result)
            video_path = os.path.dirname(lines[0]) + '.mp4'
    return render_template('video_result.html', img_result_list=img_result_list, video_path=video_path)
Ejemplo n.º 2
0
def video_result(video_time, video_name):
    record_time_path = os.path.join(app.config['UPLOAD_FOLDER'], video_time)
    record_zip_folder_path = os.path.join(record_time_path, ZIP_FOLDER)
    record_list = os.listdir(record_zip_folder_path)
    for record in record_list:
        img_result_list = []
        record_folder = os.path.join(record_zip_folder_path, record)
        print video_name
        video_file = os.path.join(record_folder, video_name.replace('mp4', 'txt'))
        video_path = ''
        with open(video_file) as fh:
            lines = fh.readlines()
            for line in lines:
                img_result = ImageResult()
                img_result.path, img_result.pre_label, img_result.score = line.strip('\n').split(' ')
                img_result.name = os.path.basename(img_result.path)
                img_result_list.append(img_result)
            video_path = os.path.dirname(lines[0]) + '.mp4'
    return render_template('video_result.html', img_result_list=img_result_list, video_path=video_path)