quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed Type on MORE_DETAILED_CAPTION (#68)

Browse Source
This commit is contained in:
John Heilman
2024-11-02 19:09:14 -08:00
committed by GitHub
parent 91c69d763f
commit 7262a579d7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
grounded_sam2_florence2_autolabel_pipeline.py
View File

@@ -17,7 +17,7 @@ Define Some Hyperparam
TASK_PROMPT = { TASK_PROMPT = {
"caption": "<CAPTION>", "caption": "<CAPTION>",
"detailed_caption": "<DETAILED_CAPTION>", "detailed_caption": "<DETAILED_CAPTION>",
"more_detailed_caption": "<MORE_DETAILED_CAPTION", "more_detailed_caption": "<MORE_DETAILED_CAPTION>",
"object_detection": "<OD>", "object_detection": "<OD>",
"dense_region_caption": "<DENSE_REGION_CAPTION>", "dense_region_caption": "<DENSE_REGION_CAPTION>",
"region_proposal": "<REGION_PROPOSAL>", "region_proposal": "<REGION_PROPOSAL>",
@@ -195,4 +195,4 @@ if __name__ == "__main__":
image_path=IMAGE_PATH image_path=IMAGE_PATH
) )
else: else:
raise NotImplementedError(f"Pipeline: {args.pipeline} is not implemented at this time") raise NotImplementedError(f"Pipeline: {args.pipeline} is not implemented at this time")