Skip to content

Commit

Permalink
Port DeepLabV3 segmentation model from keras_cv.models.legacy to th…
Browse files Browse the repository at this point in the history
…e `Task` API (keras-team#1831)

* add: DeepLabV3 backbone + preset placeholders

* add: presents for DeepLabV3 backbone

* update: minor changes

* update: minor changes

* add: license info

* update: refactored DeepLabV3 as a single Task

* add: deeplab_v3 module inside segmentation module

* update: refactor segmentation head

* add: serialization test for SegmentationHead

* remove: legacy segmentation models dir

* remove: legacy segmentation imports

* update: added weight_decay in compile()

* add: docstring for deeplabv3

* update: remove compile() and train_step() from DeepLabV3

* update: imports

* update: copyright info

* remove: Segmentation Head layer implementation + serialization test

* update: refactor DeepLabV3 segmentation head as a keras Sequential

* update: made abstractions to implement encoder-decoder architecture

* update: _make_deeplabv3_encoder

* refactor: docstrings for DeepLabV3

* refactor: docstrings for DeepLabV3

* refactor: DeepLabV3

* update: refactored DeepLabV3 to implement DeepLabV3+

* update: docstring

* update: docstring

* update: removed segmentation_head_activation

* update: removed input_shape and input_tensor

* update: remove outdated comment

* update: docstring

* update: replaced low_level_feature_layer_name with low_level_feature_pyramid_level

* update: docastring

* update: DeepLabV3

* update: remove unused imports

* update: changed DeepLabV3 to DeepLabV3+

* update: remove build

* update: docstring

* update: add docstring for dropout

* update: made encoder output upsampling dynamic

* add: tests for DeepLabV3+

* update: order of parameters

* update: example in docstring

* update: renamed feature_map

* update: docs for dropout

* update: renamed tests

* update: update tests

* update: removed test_no_nans

* update: fix linting

* update: remove cleanup_global_session

* update: tests

* update: fix linting

* update: incorporated feedback

* update: incorporated feedback

* update: incorporated feedback

* add: preset for deeplabv3+

* update: remove presets

* fix: linting

* update: make CI happy
  • Loading branch information
soumik12345 committed Jul 12, 2023
1 parent e4df665 commit c25b627
Show file tree
Hide file tree
Showing 8 changed files with 339 additions and 493 deletions.
1 change: 1 addition & 0 deletions keras_cv/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,5 +123,6 @@
from keras_cv.models.object_detection_3d.center_pillar import (
MultiHeadCenterPillar,
)
from keras_cv.models.segmentation import DeepLabV3Plus
from keras_cv.models.stable_diffusion import StableDiffusion
from keras_cv.models.stable_diffusion import StableDiffusionV2
1 change: 0 additions & 1 deletion keras_cv/models/legacy/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,6 @@
from keras_cv.models.legacy.regnet import RegNetY120
from keras_cv.models.legacy.regnet import RegNetY160
from keras_cv.models.legacy.regnet import RegNetY320
from keras_cv.models.legacy.segmentation.deeplab import DeepLabV3
from keras_cv.models.legacy.vgg16 import VGG16
from keras_cv.models.legacy.vgg19 import VGG19
from keras_cv.models.legacy.vit import ViTB16
Expand Down
327 changes: 0 additions & 327 deletions keras_cv/models/legacy/segmentation/deeplab.py

This file was deleted.

Loading

0 comments on commit c25b627

Please sign in to comment.