sun15684712291/aicheckv2
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

补全yolov5的代码

Browse Source
This commit is contained in:
sunyg
2025-03-13 17:48:59 +08:00
parent 6daea23f0a
commit 9d369b9898
42 changed files with 8571 additions and 171 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/api/business/project_detect_api.py
View File

@ -120,13 +120,20 @@ def run_detect_yolo(detect_log_in: ProjectDetectLogIn, session: Session = Depend
detect_img_count = pdc.check_detect_img(detect_log_in.detect_id, session)
if detect_img_count == 0:
return rc.response_error("推理集合中没有内容，请先到推理集合中上传图片")
detect_log = pds.run_detect_yolo(detect_log_in, detect, train, session)
thread_train = threading.Thread(target=run_event_loop, args=(detect_log.pt_url,
detect_log.folder_url,
detect_log.detect_folder_url,
detect_log.detect_version,
detect_log.id, detect_log.detect_id, session,))
thread_train.start()
if detect.file_type == 'img' or detect.file_type == 'video':
detect_log = pds.run_detect_yolo(detect_log_in, detect, train, session)
thread_train = threading.Thread(target=run_event_loop,
args=(detect_log.pt_url, detect_log.folder_url, detect_log.detect_folder_url,
detect_log.detect_version, detect_log.id, detect_log.detect_id, session,))
thread_train.start()
elif detect.file_type == 'rtsp':
if detect_log_in.pt_type == 'best':
weights_pt = train.best_pt
else:
weights_pt = train.last_pt
thread_train = threading.Thread(target=pds.run_detect_rtsp,
args=(weights_pt, detect.rtsp_url, train.train_data,))
thread_train.start()
return rc.response_success(msg="执行成功")