Commit 3cc7104d authored by zww's avatar zww
Browse files

Fix numba warning

parent 643b46d3
import warnings
import numba import numba
import numpy as np import numpy as np
from numba.errors import NumbaPerformanceWarning
from mmdet3d.core.bbox import box_np_ops from mmdet3d.core.bbox import box_np_ops
warnings.filterwarnings("ignore", category=NumbaPerformanceWarning)
@numba.njit @numba.njit
def _rotation_box2d_jit_(corners, angle, rot_mat_T): def _rotation_box2d_jit_(corners, angle, rot_mat_T):
......
...@@ -433,8 +433,8 @@ class MMDataBaseSampler(DataBaseSampler): ...@@ -433,8 +433,8 @@ class MMDataBaseSampler(DataBaseSampler):
elif blending_op == 'box': elif blending_op == 'box':
obj_mask = cv2.blur(obj_mask.astype(np.float32), (3, 3)) obj_mask = cv2.blur(obj_mask.astype(np.float32), (3, 3))
paste_mask = 1 - obj_mask paste_mask = 1 - obj_mask
img[y1:y1 + h, img[y1:y1 + h, x1:x1 +
x1:x1 + w] = (img[y1:y1 + h, x1:x1 + w].astype(np.float32) * w] = (img[y1:y1 + h, x1:x1 + w].astype(np.float32) *
paste_mask[..., None]).astype(np.uint8) paste_mask[..., None]).astype(np.uint8)
img[y1:y1 + h, x1:x1 + w] += (obj_img.astype(np.float32) * img[y1:y1 + h, x1:x1 + w] += (obj_img.astype(np.float32) *
obj_mask[..., None]).astype(np.uint8) obj_mask[..., None]).astype(np.uint8)
......
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