Skip to content

Commit 5796b76

Browse files
authored
Add BASNet Segmentation Model (#2006) (#2271)
* BASNet model initial code structure * adding test and initial preset details * adding comments * cleaning and formatting code * keras 3 support added * disabling preset test for BASNet
1 parent a6a9cc2 commit 5796b76

File tree

6 files changed

+660
-0
lines changed

6 files changed

+660
-0
lines changed

keras_cv/models/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,7 @@
190190
from keras_cv.models.object_detection.yolo_v8.yolo_v8_detector import (
191191
YOLOV8Detector,
192192
)
193+
from keras_cv.models.segmentation import BASNet
193194
from keras_cv.models.segmentation import DeepLabV3Plus
194195
from keras_cv.models.segmentation import SAMMaskDecoder
195196
from keras_cv.models.segmentation import SAMPromptEncoder

keras_cv/models/segmentation/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
from keras_cv.models.segmentation.basnet import BASNet
1516
from keras_cv.models.segmentation.deeplab_v3_plus import DeepLabV3Plus
1617
from keras_cv.models.segmentation.segformer import SegFormer
1718
from keras_cv.models.segmentation.segment_anything import SAMMaskDecoder
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
# Copyright 2023 The KerasCV Authors
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
from keras_cv.models.segmentation.basnet.basnet import BASNet

0 commit comments

Comments
 (0)