Skip to content

GitLab

Commit 6d0be0a8 authored by zww's avatar zww
Browse files

fix format

parent 3cc7104d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
mmdet3d/datasets/pipelines/dbsampler.py
View file @ 6d0be0a8
...@@ -433,9 +433,9 @@ class MMDataBaseSampler(DataBaseSampler): ...@@ -433,9 +433,9 @@ 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, x1:x1 + img[y1:y1 + h,
w] = (img[y1:y1 + h, x1:x1 + w].astype(np.float32) * x1:x1 + 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)
return img return img
......
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