Unverified Commit ba63fbdb authored by Mike Ruberry's avatar Mike Ruberry Committed by GitHub
Browse files

Updates integer division to use floor division operator (#2243)

Another instance of integer division using the division operator. In this case line 266 already shows the correct formulation, so line 185 only needs the update.
parent e6b4078e
...@@ -182,7 +182,7 @@ def _onnx_heatmaps_to_keypoints(maps, maps_i, roi_map_width, roi_map_height, ...@@ -182,7 +182,7 @@ def _onnx_heatmaps_to_keypoints(maps, maps_i, roi_map_width, roi_map_height,
pos = roi_map.reshape(num_keypoints, -1).argmax(dim=1) pos = roi_map.reshape(num_keypoints, -1).argmax(dim=1)
x_int = (pos % w) x_int = (pos % w)
y_int = ((pos - x_int) / w) y_int = ((pos - x_int) // w)
x = (torch.tensor(0.5, dtype=torch.float32) + x_int.to(dtype=torch.float32)) * \ x = (torch.tensor(0.5, dtype=torch.float32) + x_int.to(dtype=torch.float32)) * \
width_correction.to(dtype=torch.float32) width_correction.to(dtype=torch.float32)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment