Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add YOLOV8 instance segmentation model using YOLACT #2474

Merged
merged 12 commits into from
Sep 17, 2024
Merged

Add YOLOV8 instance segmentation model using YOLACT #2474

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
5446421
Add YOLOV8 instance segmentation model using YOLACT
oarriaga Aug 14, 2024
3d2e62c
Add indices to output for mask post processing selection
oarriaga Sep 6, 2024
d79112c
Add YOLOV8 instance segmentation encoder
oarriaga Sep 6, 2024
dfba5b6
Add YOLOV8 backbone features for instance segmentation model
oarriaga Sep 6, 2024
6bcef93
Add Keras copyright
oarriaga Sep 6, 2024
7ab9ae0
Update YOLOV8 instance segmentation model and loss
oarriaga Sep 6, 2024
6a5e0bd
Add yolo v8 segmentation to GPU tests
oarriaga Sep 11, 2024
0b5cbde
Fix overwrite of API for label encoder
oarriaga Sep 11, 2024
a337643
Add to API YOLOV8Segmentation model
oarriaga Sep 11, 2024
9b5a635
Fix linter E501 warnings with imports
oarriaga Sep 11, 2024
33d9014
Remove TODO
oarriaga Sep 17, 2024
5f551a3
Move docstring to class description
oarriaga Sep 17, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions keras_cv/src/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,7 @@
from keras_cv.src.models.segmentation import SAMPromptEncoder
from keras_cv.src.models.segmentation import SegmentAnythingModel
from keras_cv.src.models.segmentation import TwoWayTransformer
from keras_cv.src.models.segmentation import YOLOV8Segmentation
from keras_cv.src.models.segmentation.segformer.segformer_aliases import (
SegFormer,
)
Expand Down
3 changes: 3 additions & 0 deletions keras_cv/src/models/segmentation/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,3 +21,6 @@
SegmentAnythingModel,
)
from keras_cv.src.models.segmentation.segment_anything import TwoWayTransformer
from keras_cv.src.models.segmentation.yolo_v8_segmentation import (
YOLOV8Segmentation,
)
1 change: 1 addition & 0 deletions keras_cv/src/models/segmentation/yolo_v8_segmentation/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from .yolo_v8_segmentation import YOLOV8Segmentation
Loading
Loading