Skip to content

Commit

Permalink
Refactor action tasks and remove unused modules (#1993)
Browse files Browse the repository at this point in the history
* Refactor action tasks and remove unused modules

* Add openvino task unit test

* Add unit test for action tasks

* Update changelog

* Reflect changes from #1976

* Reflect Reviews
  • Loading branch information
jaegukhyun authored Apr 12, 2023
1 parent 10fce79 commit 159cfff
Show file tree
Hide file tree
Showing 45 changed files with 1,513 additions and 3,112 deletions.
1 change: 1 addition & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ All notable changes to this project will be documented in this file.
- Classification task refactoring (<https://github.com/openvinotoolkit/training_extensions/pull/1972>)
- Extend OTX explain CLI (<https://github.com/openvinotoolkit/training_extensions/pull/1941>)
- Segmentation task refactoring (<https://github.com/openvinotoolkit/training_extensions/pull/1977>)
- Action task refactoring (<https://github.com/openvinotoolkit/training_extensions/pull/1993>)

### Bug fixes

Expand Down
4 changes: 2 additions & 2 deletions otx/algorithms/action/adapters/mmaction/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@

from .data import OTXActionClsDataset, OTXActionDetDataset
from .models import register_action_backbones
from .utils import Exporter, patch_config, set_data_classes
from .utils import Exporter

__all__ = ["OTXActionClsDataset", "OTXActionDetDataset", "patch_config", "set_data_classes", "Exporter"]
__all__ = ["OTXActionClsDataset", "OTXActionDetDataset", "Exporter"]

register_action_backbones()
Original file line number Diff line number Diff line change
Expand Up @@ -104,6 +104,7 @@ def __init__(
):
self.otx_dataset = otx_dataset
self.labels = labels
self.CLASSES = [label.name for label in labels]
self.test_mode = test_mode
self.modality = modality

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -219,6 +219,7 @@ def __init__(
):
self.otx_dataset = otx_dataset
self.labels = labels
self.CLASSES = [label.name for label in labels]
self.test_mode = test_mode
self.modality = modality
self._FPS = fps
Expand Down
Loading

0 comments on commit 159cfff

Please sign in to comment.