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

1102

Browse Source
V0.1.0
王莹 2 years ago
parent a051f44b9d
commit 50c2c3a3a4
7 changed files with 49 additions and 58 deletions
Show Stats Download Patch File Download Diff File

20
Bank_second_part/xbank_detect_process/config_det.yaml
Unescape Escape View File

@ -1,14 +1,14 @@
detect_0:
source : /home/tp/taojx/xbank_test_videos/0711-1/0711-1_10.avi
model: ./config_phone.yaml
# detect_0:
# source : rtsp://admin:@192.168.10.203
# model: ../config_phone.yaml
# detect_1:
# source : rtsp://admin:@192.168.10.12
# model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
detect_1:
source : rtsp://admin:@192.168.10.203
model: ../config_phone.yaml
# detect_2:
# source : rtsp://admin:@192.168.10.18
# model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
# source : E:/Bank_files/Bank_03/dataset/dataset_now/phone/4/images
# model: ../config_sleep.yaml
# detect_3:
# source : rtsp://admin:@192.168.10.11
@ -27,6 +27,6 @@ detect_0:
# model: /home/yaxin/xbank/xbank_poc_test/config_phone.yaml
# detect_7:
# source : /home/yaxin/xbank/xbank_poc_test/images_test/images_del
# model: /home/yaxin/xbank/xbank_poc_test/config_sleep.yaml
# source : /home/xbank/xbank_poc_test_use/images_test/images_del

28
Bank_second_part/xbank_detect_process/config_person.yaml
Unescape Escape View File

@ -1,30 +1,30 @@
# load model file
model: /home/xbank/xbank_poc_test_use/model_file/yolov8.onnx
model_cache: /home/xbank/xbank_poc_test_use/tensort_cache/yolov8.trt
model: ./model_file/yolov8_person.onnx
model_cache: ./tensort_cache/yolov8_person.trt
# label and bbox message set
model_parameter:
device : gpu
label_names: ["person","sleep"] # model labels
device : cpu
label_names: ["person","bicycle","car",motorcycle,airplane,bus,train,truck,boat,traffic light,fire hydrant,stop sign,parking meter,bench,bird,cat, dog,horse,sheep, cow,elephant, bear,zebra,giraffe, backpack,umbrella,handbag,tie,suitcase,frisbee,skis,snowboard,sports ball,kite,baseball bat,baseball glove,skateboard,surfboard,tennis racket, bottle, wine glass,cup,fork,knife,spoon, bowl,banana,apple,sandwich,orange,broccoli,carrot,hot dog, pizza,donut,cake,chair,couch,potted plant,bed,dining table,toilet,tv,laptop,mouse,remote,keyboard,cell phone,microwave, oven,toaster,sink,refrigerator,book,clock,vase,scissors,teddy bear,hair drier,toothbrush] # model labels
# label_names: ["person"]
compara_label_names: ["person"] #
compara_relevancy: False # 'object_num'
relevancy_para : False
object_num_min : 5
confidence : 0.5
object_num_min : False # 视频中最低人数
confidence : 0.2
# save_path : /home/xbank/xbank_poc_test_use/save_path/sleep
# save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/sleep
# test_path : /home/yaxin/xbank/xbank_poc_test/test_save_path/sleep
save_path_original : False
test_path : False
save_annotations : False
save_annotations : False
save_path : False
# save videos
save_videos : /home/xbank/xbank_poc_test_use/video_save_path/person
save_videos : False
# detect time set
detect_time : 60
detect_time_small : 5
detect_ratio : 0.9
detect_time : 30
detect_time_small : 10
detect_ratio : 0.5

20
Bank_second_part/xbank_detect_process/config_phone.yaml
Unescape Escape View File

@ -1,35 +1,29 @@
# load model file
model: /home/xbank/xbank_poc_test_use/model_file/yolov5.onnx
model_cache: /home/xbank/xbank_poc_test_use/tensort_cache/yolov5.trt
model: E:/Bank_files/Bank_03/xbank_poc_test_use/model_file/yolov5_4best.onnx
model_cache: E:/Bank_files/Bank_03/xbank_poc_test_use/tensort_cache/yolov5_4best.trt
# label and bbox message set
model_parameter:
device : gpu
device : cpu
label_names: ["Keypad","hands","keyboard", "mouse","phone"] # model labels
compara_label_names: ["hands","phone"] #
compara_relevancy: overlap # 'in_bbox','overlap'
relevancy_para : 0
relevancy_para : 0.01
object_num_min : False
confidence : 0.2
# temporarily save images
# save_path : E:/Bank_files/Bank_03/xbank_poc_test_use/save_path/hands/det
# save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/hands
# test_path : E:/Bank_files/Bank_03/xbank_poc_test_use/save_path/hands/nodet
save_path : False
test_path : False
save_path_original : False
save_annotations : False
# save detect infors
save_annotations : False
# save videos
save_videos : /home/xbank/xbank_poc_test_use/video_save_path/hands
save_videos : E:/Bank_files/Bank_03/xbank_poc_test_use/video_save_path/sleep
# detect time set
detect_time : 60
detect_time_small : 5
detect_time : 30
detect_time_small : 10
detect_ratio : 0.5

20
Bank_second_part/xbank_detect_process/config_sleep.yaml
Unescape Escape View File

@ -1,31 +1,29 @@
# load model file
model: /home/xbank/xbank_poc_test_use/model_file/yolov8.onnx
model_cache: /home/xbank/xbank_poc_test_use/tensort_cache/yolov8.trt
model: ./model_file/yolov5_640_sleep.onnx
model_cache: ./tensort_cache/yolov5_640_sleep.trt
# label and bbox message set
model_parameter:
device : gpu
device : cpu
label_names: ["person","sleep"] # model labels
compara_label_names: ["sleep"] #
compara_label_names: ["sleep","person"] #
compara_relevancy: False # 'in_bbox'
relevancy_para : False
object_num_min : False
confidence : 0.5
confidence : 0.2
# save_path : /home/xbank/xbank_poc_test_use/save_path/sleep
# save_path_original : /home/yaxin/xbank/xbank_poc_test/save_path_original/sleep
# test_path : /home/yaxin/xbank/xbank_poc_test/test_save_path/sleep
save_path_original : False
test_path : False
save_annotations : False
save_path : False
# save videos
save_videos : /home/xbank/xbank_poc_test_use/video_save_path/sleep
save_videos : ./video_save_path/sleep
# detect time set
detect_time : 60
detect_time_small : 5
detect_time : 30
detect_time_small : 10
detect_ratio : 0.5

9
Bank_second_part/xbank_detect_process/main.py
Unescape Escape View File

@ -4,7 +4,6 @@ from analysis_data.config_load import get_configs
from multiprocessing import Pool
from loguru import logger
import os
import time
# 日志配置
log_path = os.path.join(__file__, "../logs/xbank.log")
@ -28,16 +27,16 @@ def get_args_list(args_data):
return args_list
def start_worker(args_list):
def start_worker(args_list_n):
while True:
try:
logger.info(f" process {args_list} crashed. Starting...")
logger.info(f" process {args_list_n} crashed. Starting...")
data_load(args_list)
data_load(args_list_n)
except Exception as e:
logger.info(f" process {args_list} crashed. Restarting...")
logger.info(f" process {args_list_n} crashed. Restarting...")
logger.debug(e)

8
Bank_second_part/xbank_detect_process/main_process.py
Unescape Escape View File

@ -207,8 +207,8 @@ def rtsp_detect_process(source, model_data, model_inference):
time_list=update_det_fps, video_path=save_video, fps=fps)
result_path = json_save(re_list, json_path)
send_message(update_det_fps=update_det_fps, result_path=result_path,
source=source, result_lables=result_lables)
# send_message(update_det_fps=update_det_fps, result_path=result_path,
# source=source, result_lables=result_lables)
else:
# print(video_path)
@ -349,7 +349,7 @@ def save_process(images, determine_bbox, model_data):
img_save = drawing_frame(
images_frame=images['frame'], result_list=determine_bbox)
images.update({"frame": img_save})
images.update({"img_save": img_save})
if model_data["save_path"]:
@ -396,7 +396,7 @@ def images_save(images, save_path):
full_name = os.path.join(save_path, images_name)
cv2.imwrite(full_name, images['frame'])
cv2.imwrite(full_name, images['img_save'])
return full_name

2
Bank_second_part/xbank_detect_process/readme.md
Unescape Escape View File

@ -1,6 +1,6 @@
1、主程序运行main.py文件
2、main.py文件中args加载检测配置文件config_det.yaml，为了保证路径没问题，请设置绝对路径
2、main.py文件中args加载检测配置文件config_det.yaml，这部分更改了使用相对路径，请按照示例格式设置路径
3、config_det.yaml中配置

Write Preview
Loading…
Cancel
Save