Unverified 提交 b884ea36 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Simplify TF normalized to pixels (#6494)

上级 6445a813
......@@ -446,10 +446,7 @@ class DetectMultiBackend(nn.Module):
if int8:
scale, zero_point = output['quantization']
y = (y.astype(np.float32) - zero_point) * scale # re-scale
y[..., 0] *= w # x
y[..., 1] *= h # y
y[..., 2] *= w # w
y[..., 3] *= h # h
y[..., :4] *= [w, h, w, h] # xywh normalized to pixels
y = torch.tensor(y) if isinstance(y, np.ndarray) else y
return (y, []) if val else y
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论