mirror of
https://github.com/huchenlei/Depth-Anything.git
synced 2026-01-26 15:29:46 +00:00
30 lines
892 B
Python
30 lines
892 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
from typing import Any, Tuple
|
|
|
|
from torch.utils.data import Dataset
|
|
|
|
|
|
class DatasetWithEnumeratedTargets(Dataset):
|
|
def __init__(self, dataset):
|
|
self._dataset = dataset
|
|
|
|
def get_image_data(self, index: int) -> bytes:
|
|
return self._dataset.get_image_data(index)
|
|
|
|
def get_target(self, index: int) -> Tuple[Any, int]:
|
|
target = self._dataset.get_target(index)
|
|
return (index, target)
|
|
|
|
def __getitem__(self, index: int) -> Tuple[Any, Tuple[Any, int]]:
|
|
image, target = self._dataset[index]
|
|
target = index if target is None else target
|
|
return image, (index, target)
|
|
|
|
def __len__(self) -> int:
|
|
return len(self._dataset)
|