chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-05 18:40:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
Zhijie Lin 2023-05-25 17:13:50 +08:00
parent 2d914f561d
commit ccf3db9207
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GroundingModel.py
View File

@ -35,7 +35,7 @@ class GroundingModule(nn.Module):
groundingdino_config_file = "./groundingdino/config/GroundingDINO_SwinT_OGC.py" groundingdino_config_file = "./groundingdino/config/GroundingDINO_SwinT_OGC.py"
self.grounding_model = load_groundingdino_model(groundingdino_config_file, groundingdino_checkpoint).to(device) self.grounding_model = load_groundingdino_model(groundingdino_config_file, groundingdino_checkpoint).to(device)
self.sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint)).to(device) self.sam_predictor = SamPredictor(build_sam(checkpoint=sam_checkpoint).to(device))
def prompt2mask(self, original_image, prompt, box_threshold=0.25, text_threshold=0.25, num_boxes=10): def prompt2mask(self, original_image, prompt, box_threshold=0.25, text_threshold=0.25, num_boxes=10):
def image_transform_grounding(init_image): def image_transform_grounding(init_image):