File tree Expand file tree Collapse file tree 2 files changed +10
-11
lines changed Expand file tree Collapse file tree 2 files changed +10
-11
lines changed Original file line number Diff line number Diff line change 23
23
version = "1.0.0" ,
24
24
)
25
25
class GroundingDinoInvocation (BaseInvocation ):
26
- """Runs a Grounding DINO model (https://arxiv.org/pdf/2303.05499). Performs zero-shot bounding-box object detection
27
- from a text prompt.
26
+ """Runs a Grounding DINO model. Performs zero-shot bounding-box object detection from a text prompt."""
28
27
29
- Reference:
30
- - https://huggingface.co/docs/transformers/v4.43.3/en/model_doc/grounding-dino#grounded-sam
31
- - https://github.com/NielsRogge/Transformers-Tutorials/blob/a39f33ac1557b02ebfb191ea7753e332b5ca933f/Grounding%20DINO/GroundingDINO_with_Segment_Anything.ipynb
32
- """
28
+ # Reference:
29
+ # - https://arxiv.org/pdf/2303.05499
30
+ # - https://huggingface.co/docs/transformers/v4.43.3/en/model_doc/grounding-dino#grounded-sam
31
+ # - https://github.com/NielsRogge/Transformers-Tutorials/blob/a39f33ac1557b02ebfb191ea7753e332b5ca933f/Grounding%20DINO/GroundingDINO_with_Segment_Anything.ipynb
33
32
34
33
prompt : str = InputField (description = "The prompt describing the object to segment." )
35
34
image : ImageField = InputField (description = "The image to segment." )
Original file line number Diff line number Diff line change 26
26
version = "1.0.0" ,
27
27
)
28
28
class SegmentAnythingInvocation (BaseInvocation ):
29
- """Runs a Segment Anything Model (https://arxiv.org/pdf/2304.02643).
29
+ """Runs a Segment Anything Model."""
30
30
31
- Reference:
32
- - https://huggingface.co/docs/transformers/v4.43.3/en/model_doc/grounding-dino#grounded-sam
33
- - https://github.com/NielsRogge/Transformers-Tutorials/blob/a39f33ac1557b02ebfb191ea7753e332b5ca933f/Grounding%20DINO/GroundingDINO_with_Segment_Anything.ipynb
34
- """
31
+ # Reference:
32
+ # - https://arxiv.org/pdf/2304.02643
33
+ # - https://huggingface.co/docs/transformers/v4.43.3/en/model_doc/grounding-dino#grounded-sam
34
+ # - https://github.com/NielsRogge/Transformers-Tutorials/blob/a39f33ac1557b02ebfb191ea7753e332b5ca933f/Grounding%20DINO/GroundingDINO_with_Segment_Anything.ipynb
35
35
36
36
image : ImageField = InputField (description = "The image to segment." )
37
37
bounding_boxes : list [BoundingBoxField ] = InputField (description = "The bounding boxes to prompt the SAM model with." )
You can’t perform that action at this time.
0 commit comments