if test_mode == 'validation': video_dict = val_dict else: video_dict = test_dict batch_video_list = data_loader.getBatchListTest(video_dict, batch_size, shuffle=False) batch_result_xmin = [] batch_result_xmax = [] batch_result_iou = [] batch_result_pstart = [] batch_result_pend = [] print('Runing DBG model ...') for idx in tqdm.tqdm(range(len(batch_video_list))): batch_anchor_xmin, batch_anchor_xmax, batch_anchor_feature = data_loader.getProposalDataTest( batch_video_list[idx], video_dict) out_iou, out_start, out_end = sess.run([iou_mat, pstart, pend], feed_dict={X_feature: batch_anchor_feature}) batch_result_xmin.append(batch_anchor_xmin) batch_result_xmax.append(batch_anchor_xmax) batch_result_iou.append(out_iou[:, :, :, 0]) batch_result_pstart.append(out_start[:, :, 0]) batch_result_pend.append(out_end[:, :, 0]) print('Saving results ...') columns = ["iou", "start", "end", "xmin", "xmax"] for idx in tqdm.tqdm(range(len(batch_video_list))): b_video = batch_video_list[idx] b_xmin = batch_result_xmin[idx] b_xmax = batch_result_xmax[idx]
batch_video_list = data_loader.getBatchListTest(video_dict, batch_size) """ init result list """ batch_result_xmin = [] batch_result_xmax = [] batch_result_iou = [] batch_result_pstart = [] batch_result_pend = [] """ Run DBG model """ print('Runing DBG model ...') for idx in mmcv.track_iter_progress(range(len(batch_video_list))): """ Get batch data """ batch_anchor_xmin, batch_anchor_xmax, batch_anchor_feature = \ data_loader.getProposalDataTest( batch_video_list[idx], dbg_config) """ Run batch data """ out_iou, out_start, out_end = sess.run( [iou_mat, pstart, pend], feed_dict={X_feature: batch_anchor_feature}) batch_result_xmin.append(batch_anchor_xmin) batch_result_xmax.append(batch_anchor_xmax) batch_result_iou.append(out_iou[:, :, :, 0]) batch_result_pstart.append(out_start[:, :, 0]) batch_result_pend.append(out_end[:, :, 0]) utils.save_proposals_result(batch_video_list, batch_result_xmin, batch_result_xmax, batch_result_iou, batch_result_pstart, batch_result_pend, tscale, result_dir)