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

Move invokeai/backend/grounded_sam -> invokeai/backend/image_util/grounded_sam

Browse Source
This commit is contained in:
Ryan Dick 2024-07-31 10:00:30 -04:00
parent bcd1483a14
commit 9f448fecb7
7 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/grounded_sam.py
View File

@ -14,10 +14,10 @@ from invokeai.app.invocations.baseinvocation import BaseInvocation, invocation
from invokeai.app.invocations.fields import ImageField, InputField
from invokeai.app.invocations.primitives import ImageOutput
from invokeai.app.services.shared.invocation_context import InvocationContext
from invokeai.backend.grounded_sam.detection_result import DetectionResult
from invokeai.backend.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline
from invokeai.backend.grounded_sam.mask_refinement import mask_to_polygon, polygon_to_mask
from invokeai.backend.grounded_sam.segment_anything_model import SegmentAnythingModel
from invokeai.backend.image_util.grounded_sam.detection_result import DetectionResult
from invokeai.backend.image_util.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline
from invokeai.backend.image_util.grounded_sam.mask_refinement import mask_to_polygon, polygon_to_mask
from invokeai.backend.image_util.grounded_sam.segment_anything_model import SegmentAnythingModel
GROUNDING_DINO_MODEL_ID = "IDEA-Research/grounding-dino-tiny"
SEGMENT_ANYTHING_MODEL_ID = "facebook/sam-vit-base"

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

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

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

@ -4,7 +4,7 @@ import torch
from PIL import Image
from transformers.pipelines import ZeroShotObjectDetectionPipeline
from invokeai.backend.grounded_sam.detection_result import DetectionResult
from invokeai.backend.image_util.grounded_sam.detection_result import DetectionResult
class GroundingDinoPipeline:

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

2
invokeai/backend/grounded_sam/segment_anything_model.py → invokeai/backend/image_util/grounded_sam/segment_anything_model.py
View File

@ -5,7 +5,7 @@ from PIL import Image
from transformers.models.sam import SamModel
from transformers.models.sam.processing_sam import SamProcessor
from invokeai.backend.grounded_sam.detection_result import DetectionResult
from invokeai.backend.image_util.grounded_sam.detection_result import DetectionResult
class SegmentAnythingModel:

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 transformers import CLIPTokenizer
from invokeai.backend.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline
from invokeai.backend.grounded_sam.segment_anything_model import SegmentAnythingModel
from invokeai.backend.image_util.grounded_sam.grounding_dino_pipeline import GroundingDinoPipeline
from invokeai.backend.image_util.grounded_sam.segment_anything_model import SegmentAnythingModel
from invokeai.backend.ip_adapter.ip_adapter import IPAdapter
from invokeai.backend.lora import LoRAModelRaw
from invokeai.backend.model_manager.config import AnyModel