Feature/sg 1277 depth estimation dataset support #1705

shaydeci · 2023-12-14T09:27:05Z

Base class for depth estimation dataset.
NYUv2 Depth Estimaiton dataset class + subset of examples for testing.
DepthEstimationSample + PipelineAdapteor support for it.
Unit tests and Albumentations integration test.

…th_estimation_data

src/super_gradients/training/samples/depth_estimation_sample.py

ofrimasad

very nice.
a few minor comments

src/super_gradients/common/object_names.py

...r_gradients/training/datasets/depth_estimation_datasets/abstract_depth_estimation_dataset.py

src/super_gradients/training/transforms/depth_estimation/abstract_depth_estimation_transform.py

...r_gradients/training/datasets/depth_estimation_datasets/abstract_depth_estimation_dataset.py

src/super_gradients/training/utils/visualization/depth_estimation.py

...r_gradients/training/datasets/depth_estimation_datasets/abstract_depth_estimation_dataset.py

BloodAxe

LGTM

* added sample, datasets and transforms * added docsand instructions to dataset * dataset examples and imports * added plot * pipeline adaptor for albu added * added albu test * modified object names to be more clear * removed apply transform * added depth visualization * minor improvements for visualization module * fixed getitem returning sample * fixed getitem return type * fixed getitem redundant clipping * fixed plot ignore * fixed unit test * added clarifications in getitem and load_sample * updated plot() * updated casting image to fp32

* all code squashed * add to Makefile and update Readme * remove ppritn and fix pip install * update with all cell outputs * Feature/sg 1277 depth estimation dataset support (#1705) * added sample, datasets and transforms * added docsand instructions to dataset * dataset examples and imports * added plot * pipeline adaptor for albu added * added albu test * modified object names to be more clear * removed apply transform * added depth visualization * minor improvements for visualization module * fixed getitem returning sample * fixed getitem return type * fixed getitem redundant clipping * fixed plot ignore * fixed unit test * added clarifications in getitem and load_sample * updated plot() * updated casting image to fp32 * update notebook to 3.5 --------- Co-authored-by: Eugene Khvedchenya <ekhvedchenya@gmail.com> Co-authored-by: Shay Aharon <80472096+shaydeci@users.noreply.github.com>

shaydeci added 7 commits December 13, 2023 16:53

added sample, datasets and transforms

81e6e46

added docsand instructions to dataset

d1854fa

dataset examples and imports

2fc0d23

added plot

af20d4e

Merge remote-tracking branch 'origin/master' into feature/SG-1277_dep…

8da4284

…th_estimation_data

pipeline adaptor for albu added

091796d

added albu test

0459f62

shaydeci requested review from ofrimasad, BloodAxe and Louis-Dupont as code owners December 14, 2023 09:27

shaydeci changed the title ~~Feature/sg 1277 depth estimation data~~ Feature/sg 1277 depth estimation dataset support Dec 14, 2023

BloodAxe reviewed Dec 14, 2023

View reviewed changes

src/super_gradients/training/samples/depth_estimation_sample.py Show resolved Hide resolved

ofrimasad reviewed Dec 17, 2023

View reviewed changes

shaydeci and others added 3 commits December 18, 2023 10:09

modified object names to be more clear

2541c95

removed apply transform

dddf9d3

Merge branch 'master' into feature/SG-1277_depth_estimation_data

a1d0e37