Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Split invokeai/backend/image_util/segment_anything/ dir into grounding_dino/ and segment_anything/

Browse Source
This commit is contained in:
Ryan Dick 2024-07-31 12:28:47 -04:00
parent 0193267a53
commit fca119773b
9 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/invocations/grounding_dino.py
View File

@ -9,8 +9,8 @@ from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
from invokeai.app.invocations.fields import BoundingBoxField, ImageField, InputField from invokeai.app.invocations.fields import BoundingBoxField, ImageField, InputField
from invokeai.app.invocations.primitives import BoundingBoxCollectionOutput from invokeai.app.invocations.primitives import BoundingBoxCollectionOutput
from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.app.services.shared.invocation_context import InvocationContext
from invokeai.backend.image_util.grounded_sam.detection_result import DetectionResult from invokeai.backend.image_util.grounding_dino.detection_result import DetectionResult
from invokeai.backend.image_util.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline from invokeai.backend.image_util.grounding_dino.grounding_dino_pipeline import GroundingDinoPipeline
GROUNDING_DINO_MODEL_ID = "IDEA-Research/grounding-dino-tiny" GROUNDING_DINO_MODEL_ID = "IDEA-Research/grounding-dino-tiny"

4
invokeai/app/invocations/segment_anything_model.py
View File

@ -13,8 +13,8 @@ from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
from invokeai.app.invocations.fields import BoundingBoxField, ImageField, InputField from invokeai.app.invocations.fields import BoundingBoxField, ImageField, InputField
from invokeai.app.invocations.primitives import ImageOutput from invokeai.app.invocations.primitives import ImageOutput
from invokeai.app.services.shared.invocation_context import InvocationContext from invokeai.app.services.shared.invocation_context import InvocationContext
from invokeai.backend.image_util.grounded_sam.mask_refinement import mask_to_polygon, polygon_to_mask from invokeai.backend.image_util.segment_anything.mask_refinement import mask_to_polygon, polygon_to_mask
from invokeai.backend.image_util.grounded_sam.segment_anything_model import SegmentAnythingModel from invokeai.backend.image_util.segment_anything.segment_anything_model import SegmentAnythingModel
SEGMENT_ANYTHING_MODEL_ID = "facebook/sam-vit-base" SEGMENT_ANYTHING_MODEL_ID = "facebook/sam-vit-base"

0
invokeai/backend/image_util/grounded_sam/__init__.py → invokeai/backend/image_util/grounding_dino/__init__.py
View File

0
invokeai/backend/image_util/grounded_sam/detection_result.py → invokeai/backend/image_util/grounding_dino/detection_result.py
View File

2
invokeai/backend/image_util/grounded_sam/grounding_dino_pipeline.py → invokeai/backend/image_util/grounding_dino/grounding_dino_pipeline.py
View File

@ -4,7 +4,7 @@ import torch
from PIL import Image from PIL import Image
from transformers.pipelines import ZeroShotObjectDetectionPipeline from transformers.pipelines import ZeroShotObjectDetectionPipeline
from invokeai.backend.image_util.grounded_sam.detection_result import DetectionResult from invokeai.backend.image_util.grounding_dino.detection_result import DetectionResult
from invokeai.backend.raw_model import RawModel from invokeai.backend.raw_model import RawModel

0
invokeai/backend/image_util/segment_anything/__init__.py Normal file
View File

0
invokeai/backend/image_util/grounded_sam/mask_refinement.py → invokeai/backend/image_util/segment_anything/mask_refinement.py
View File

0
invokeai/backend/image_util/grounded_sam/segment_anything_model.py → invokeai/backend/image_util/segment_anything/segment_anything_model.py
View File

4
invokeai/backend/model_manager/load/model_util.py
View File

@ -11,8 +11,8 @@ from diffusers.pipelines.pipeline_utils import DiffusionPipeline
from diffusers.schedulers.scheduling_utils import SchedulerMixin from diffusers.schedulers.scheduling_utils import SchedulerMixin
from transformers import CLIPTokenizer from transformers import CLIPTokenizer
from invokeai.backend.image_util.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline from invokeai.backend.image_util.grounding_dino.grounding_dino_pipeline import GroundingDinoPipeline
from invokeai.backend.image_util.grounded_sam.segment_anything_model import SegmentAnythingModel from invokeai.backend.image_util.segment_anything.segment_anything_model import SegmentAnythingModel
from invokeai.backend.ip_adapter.ip_adapter import IPAdapter from invokeai.backend.ip_adapter.ip_adapter import IPAdapter
from invokeai.backend.lora import LoRAModelRaw from invokeai.backend.lora import LoRAModelRaw
from invokeai.backend.model_manager.config import AnyModel from invokeai.backend.model_manager.config import AnyModel