From 2dc7ed1287a211fdd135f184a6c33f1716329411 Mon Sep 17 00:00:00 2001 From: MrKarimiD Date: Thu, 5 Jun 2025 13:54:13 -0400 Subject: [PATCH] change the mask to roi --- lantern_scripts/mask_humans2.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/lantern_scripts/mask_humans2.py b/lantern_scripts/mask_humans2.py index 02eb22743..c15c7f007 100644 --- a/lantern_scripts/mask_humans2.py +++ b/lantern_scripts/mask_humans2.py @@ -82,15 +82,23 @@ if __name__ == '__main__': image_np_rot = (original_image.data * 255).astype(np.uint8) if folder['mask_person']: - text_prompt = "person" + text_prompt = "person head torso arms legs feet shoes" # 90 degrees rotat # image_np_rot = original_image_np # np.rot90(original_image_np, k=folder['flip_k']) masks, boxes, phrases, logits = model.predict(Image.fromarray(image_np_rot), text_prompt) masks = masks.detach().cpu().numpy() + boxes = boxes.detach().cpu().numpy() if masks.size != 0: # masks = np.rot90(masks, k=-folder['flip_k'], axes=(1, 2)) + + photographer_mask = np.zeros((masks.shape[1], masks.shape[2], 1), dtype=np.uint8) + for i in range(boxes.shape[0]): + color = (255, 255, 255) # White color in BGR + cv2.rectangle(photographer_mask, (int(boxes[i, 0]), int(boxes[i, 1])), (int(boxes[i, 2]), int(boxes[i, 3])), (255, 255, 255), cv2.FILLED) + - mask_skylibs = EnvironmentMap(np.transpose(masks, (1, 2, 0)).astype(np.uint8), 'latlong') + # mask_skylibs = EnvironmentMap(np.transpose(masks, (1, 2, 0)).astype(np.uint8), 'latlong') + mask_skylibs = EnvironmentMap(photographer_mask, 'latlong') inv_dcm = rotation_matrix(azimuth=0, elevation=0, roll=-np.pi/2) mask_skylibs.rotate(inv_dcm) masks = mask_skylibs.data