UPDATE openvino

This commit is contained in:
HJW
2024-09-11 14:16:15 +08:00
parent 7fab3b3ffa
commit 2b20732119
2 changed files with 160 additions and 160 deletions

View File

@ -54,7 +54,6 @@ class Detection:
flag, det_cpu, dst_img, masks, category_names = self.model.model_inference(img, 0)
else:
flag, det_cpu, scores, masks, category_names = self.model.segment_objects(img)
dst_img = img
if flag == 1:
xyz = []
nx_ny_nz = []
@ -132,6 +131,9 @@ class Detection:
select_point = np.array(select_point)
pm_seg = select_point.reshape(-1, 3)
pm_seg = pm_seg[~np.isnan(pm_seg).all(axis=-1), :] # 剔除 nan
if pm_seg.size < 300:
print("分割点云数量较少,无法拟合平面")
continue
# cv2.imshow('result', piont_result)
'''