|
|
@ -48,7 +48,6 @@ class Model(ModelBase): |
|
|
|
output_torch = torch.tensor(output) |
|
|
|
boxout = nms(output_torch, conf_thres=0.4,iou_thres=0.5) |
|
|
|
del output_torch |
|
|
|
output_torch = None |
|
|
|
pred_all = boxout[0].numpy() # 转换为numpy数组 -- [[],[],[]] --[x1,y1,x2,y2,置信度,coco_index] |
|
|
|
# pred_all[:, :4] 取所有行的前4列,pred_all[:,1]--第一列 |
|
|
|
scale_coords([640, 640], pred_all[:, :4], image.shape, ratio_pad=(scale_ratio, pad_size)) # 将推理结果缩放到原始图片大小 |
|
|
|