-
Notifications
You must be signed in to change notification settings - Fork 465
Bug during RGB converting in data pipeline #2584
Copy link
Copy link
Closed
Labels
BUGSomething isn't workingSomething isn't working
Description
Describe the bug
When creating Image from datumato media, the media has been already converted to RGB as described in both cases
Imagefrom path: https://github.com/openvinotoolkit/training_extensions/blob/b4fcc6d055c27792d4769ee1ee343eb2b9bfea7e/src/otx/api/entities/image.py#L94Imagefrom data: https://github.com/openvinotoolkit/training_extensions/blob/b4fcc6d055c27792d4769ee1ee343eb2b9bfea7e/src/otx/core/data/adapter/base_dataset_adapter.py#L412
But this has been again converted to RGB in data pipeline, for instances, in classification tasks: https://github.com/openvinotoolkit/training_extensions/blob/b4fcc6d055c27792d4769ee1ee343eb2b9bfea7e/src/otx/algorithms/classification/configs/base/data/data_pipeline.py#L8
As a result, there are some performance drop with current data pipelines..
| to_rgb | True (as-is) | False (to-be) |
|---|---|---|
| CIFAR100 (#2500 training data) | Acc: 0.423 / E2E time: 120.16 sec | Acc: 0.473 / E2E time: 159.17 sec |
| CUB200 (#1200 training data) | Acc: 0.615 / E2E time: 41.27 sec | Acc: 0.680 / E2E time: 41.92 sec |
Steps to Reproduce
- Turn
to_rgb=False - Do experiments
Environment:
- OS: Ubuntu 20.04
- Framework version: PyTorch 2.0
- Python version: Python 3.10
- OpenVINO version:
- CUDA/cuDNN version: 11.7
- GPU model and memory: RTX3090
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
BUGSomething isn't workingSomething isn't working