def det_img(model_inference,images_frame,confidence,label_name_list): result = model_inference.predict(images_frame) bbox_list = result.boxes labels_ids_list = result.label_ids scores_list = result.scores result_list = [] for i in range(len(labels_ids_list)): if scores_list[i] > confidence: if int(labels_ids_list[i]) in range(len(label_name_list)): result_dict = {label_name_list[int(labels_ids_list[i])]:bbox_list[i]} result_list.append(result_dict) return result_list