Yaxin
/
XZNSH-Code-AI
31
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新代码逻辑

Browse Source
V0.1.0
王莹 2 years ago
parent af2a1868d4
commit 8d57b523d8
1 changed files with 83 additions and 73 deletions
Show Stats Download Patch File Download Diff File

48
Bank_second_part/detect_process/video_process.py
Unescape Escape View File

@ -111,7 +111,6 @@ class DealVideo():
# print('video_fps:',video_fps,'count_fps:',count_fps) # print('video_fps:',video_fps,'count_fps:',count_fps)
break break
# print('count_fps_read_video=',count_fps) # print('count_fps_read_video=',count_fps)
imgsize = frame.shape imgsize = frame.shape
@ -139,13 +138,13 @@ class DealVideo():
count_fps_del_re,updata_result_contact = self.analysis_by_bbox(imgsize=imgsize, count_fps_del_re,updata_result_contact = self.analysis_by_bbox(imgsize=imgsize,
detect_result=person_list, detect_result=person_list,
dertpara=10, dertpara=4,
start_fps=start_fps, start_fps=start_fps,
now_fps=count_fps, now_fps=count_fps,
label_name='person', label_name='person',
video_path=video_path, video_path=video_path,
frame_result_contact=frame_result_contact, frame_result_contact=frame_result_contact,
parameter_fps=50, parameter_fps=150,
count_fps_del=count_fps_del, count_fps_del=count_fps_del,
video_end=video_end video_end=video_end
) )
@ -220,6 +219,9 @@ class DealVideo():
start_fps = head_result_contact[0]['fps'] start_fps = head_result_contact[0]['fps']
else: else:
start_fps = count_fps start_fps = count_fps
count_fps_del_re,updata_result_contact = self.analysis_by_bbox(imgsize=imgsize, count_fps_del_re,updata_result_contact = self.analysis_by_bbox(imgsize=imgsize,
detect_result=head_result_1, detect_result=head_result_1,
dertpara=1, dertpara=1,
@ -228,7 +230,7 @@ class DealVideo():
label_name='head', label_name='head',
video_path=video_path, video_path=video_path,
frame_result_contact=head_result_contact, frame_result_contact=head_result_contact,
parameter_fps=25, parameter_fps=80,
count_fps_del=count_fps_del_head, count_fps_del=count_fps_del_head,
video_end=video_end video_end=video_end
) )
@ -250,12 +252,14 @@ class DealVideo():
label_name='hands', label_name='hands',
video_path=video_path, video_path=video_path,
frame_result_contact=hands_result_contact, frame_result_contact=hands_result_contact,
parameter_fps=25, parameter_fps=80,
count_fps_del=count_fps_del_hand, count_fps_del=count_fps_del_hand,
video_end=video_end video_end=video_end
) )
count_fps_del_hand = count_fps_del_re count_fps_del_hand = count_fps_del_re
hands_result_contact = updata_result_contact hands_result_contact = updata_result_contact
# print(count_fps,'----------------hands_result_1-------------------------------',hands_result_1)
# print(count_fps,"-------------------------updata_result_contact----------------------:",updata_result_contact)
count_fps += 1 count_fps += 1
@ -277,6 +281,8 @@ class DealVideo():
predictor=self.predictor, predictor=self.predictor,
InferenceHelper=self.infer) InferenceHelper=self.infer)
if result_list['topk_scores'] > 0.9:
video_base_name = os.path.basename(video_path) video_base_name = os.path.basename(video_path)
video_save_select_path = self.video_save_file + '/' + 'video_select_dect/'+ str(result_list['topk_class']) video_save_select_path = self.video_save_file + '/' + 'video_select_dect/'+ str(result_list['topk_class'])
os.makedirs(video_save_select_path, exist_ok=True) os.makedirs(video_save_select_path, exist_ok=True)
@ -295,7 +301,7 @@ class DealVideo():
''' '''
imgsize:图片的尺寸, imgsize:图片的尺寸,
detect_result:检测到的图像的结果,bboxlist detect_result:检测到的图像的结果,bboxlist
dertpara:标注框修正参数,取整数,2,3 dertpara:scale_factor：缩放因子，大于 1 表示放大，小于 1 表示缩小
start_fps: 对比列表中的起始帧 start_fps: 对比列表中的起始帧
now_fpsl:当前帧率 now_fpsl:当前帧率
label_name:用于分析的检测类别 label_name:用于分析的检测类别
@ -306,9 +312,7 @@ class DealVideo():
''' '''
bbox_list = tools_function.para_list_correction(images_size=imgsize, bbox_list = tools_function.para_list_correction(images_size=imgsize,bbox_list=detect_result,dertpara=dertpara)
bbox_list=detect_result,
dertpara=dertpara)
count_fps_del_re,update_frame_result_contact = self.get_cut_message(fps1=now_fps, count_fps_del_re,update_frame_result_contact = self.get_cut_message(fps1=now_fps,
label_name = label_name, label_name = label_name,
@ -354,20 +358,20 @@ class DealVideo():
# cut_dict = {'video_path':video_path,'label_name':label_name,"stop_fps":fps1,'bbox_list':example_lst} # cut_dict = {'video_path':video_path,'label_name':label_name,"stop_fps":fps1,'bbox_list':example_lst}
# start_fps = example_lst[0]['fps'] start_fps = example_lst[0]['fps']
if count_fps_del <= 2: if count_fps_del <= 5:
frame_result_contact = frame_result_contact frame_result_contact = frame_result_contact
count_fps_del = count_fps_del + 1 count_fps_del = count_fps_del + 1
# else: # else:
# if (fps1 - start_fps) < 5: if (fps1 - start_fps) < 15:
# frame_result_contact = frame_result_contact frame_result_contact = frame_result_contact
else:
else:
cut_dict = {'video_path':video_path,'label_name':label_name,"stop_fps":fps1,'bbox_list':example_lst} cut_dict = {'video_path':video_path,'label_name':label_name,"stop_fps":fps1,'bbox_list':example_lst}
frame_result_contact = [item for item in frame_result_contact if item not in example_lst] frame_result_contact = [item for item in frame_result_contact if item not in example_lst]
self.cutbboxQueue.put(cut_dict) self.cutbboxQueue.put(cut_dict)
@ -516,7 +520,10 @@ class DealVideo():
bbox_int_list = [int(bbox_list[0]),int(bbox_list[1]),int(bbox_list[2]),int(bbox_list[3])] bbox_int_list = [int(bbox_list[0]),int(bbox_list[1]),int(bbox_list[2]),int(bbox_list[3])]
w = bbox_int_list[2] - bbox_int_list[0] w = bbox_int_list[2] - bbox_int_list[0]
h = bbox_int_list[3] - bbox_int_list[1] h = bbox_int_list[3] - bbox_int_list[1]
size = [w,h]
if tools_function.determine_zero(size):
size = (w,h) size = (w,h)
# 根据标签保存不同视频分类 # 根据标签保存不同视频分类
# bbox_name = '{}-{}-{}_{}'.format(int(bbox_list[0]), int(bbox_list[1]), int(bbox_list[2]), int(bbox_list[3])) # bbox_name = '{}-{}-{}_{}'.format(int(bbox_list[0]), int(bbox_list[1]), int(bbox_list[2]), int(bbox_list[3]))
video_name_save = '{}__{}-{}-{}_{}-{}-{}-{}.avi'.format(video_basename, start_fps, stop_fps, video_fps,int(bbox_list[0]), int(bbox_list[1]), int(bbox_list[2]), int(bbox_list[3])) video_name_save = '{}__{}-{}-{}_{}-{}-{}-{}.avi'.format(video_basename, start_fps, stop_fps, video_fps,int(bbox_list[0]), int(bbox_list[1]), int(bbox_list[2]), int(bbox_list[3]))
@ -534,6 +541,9 @@ class DealVideo():
videoWriter.release() videoWriter.release()
self.videoQueue2.put(video_save_path) self.videoQueue2.put(video_save_path)
else:
print('-----------------agrag-----------------',size,'-----------------agag-----------------')
cap.release() cap.release()
else: else:
@ -560,8 +570,8 @@ class DealVideo():
self.videoQueue3.put(video_path) self.videoQueue3.put(video_path)
if labels == 'head' or labels == 'hands': # if labels == 'head' or labels == 'hands':
if labels == 'hands':
self.videodetQueue.put(video_path) self.videodetQueue.put(video_path)
else: else:
@ -581,7 +591,7 @@ class DealVideo():
if __name__ == '__main__': if __name__ == '__main__':
t1 = time.time() t1 = time.time()
video = "test_video/test/0711ahgh.avi" video = "E:/Bank_files/Bank_02/dataset/video_kf/02.mp4"
video_save = 'test_video' video_save = 'test_video'
# 初始化目标检测 # 初始化目标检测
@ -589,8 +599,8 @@ if __name__ == '__main__':
# 初始化pptsmv2 # 初始化pptsmv2
config = 'model_file/inference/pptsm_lcnet_k400_16frames_uniform.yaml' # 配置文件地址 config = 'model_file/inference/pptsm_lcnet_k400_16frames_uniform.yaml' # 配置文件地址
model_file = 'model_file/inference/ppTSMv2.pdmodel' # 推理模型存放地址 model_file = 'model_file/inference_hands_2/ppTSMv2.pdmodel' # 推理模型存放地址
params_file = 'model_file/inference/ppTSMv2.pdiparams' params_file = 'model_file/inference_hands_2/ppTSMv2.pdiparams'
# batch_size= 1 # batch_size= 1
infer,predictor = PP_TSMv2_predict().create_inference_model(config,model_file,params_file) infer,predictor = PP_TSMv2_predict().create_inference_model(config,model_file,params_file)
# PP_TSMv2_predict().predict(config,input_file,batch_size,predictor,infer) # PP_TSMv2_predict().predict(config,input_file,batch_size,predictor,infer)

Write Preview
Loading…
Cancel
Save