spikingjelly.datasets.hardvs 源代码
import multiprocessing
import shutil
from concurrent.futures import ThreadPoolExecutor
from pathlib import Path
from typing import Callable, Optional, Tuple
from torchvision.datasets.utils import extract_archive
from .base import NeuromorphicDatasetFolder
__all__ = ["HARDVS"]
[文档]
class HARDVS(NeuromorphicDatasetFolder):
def __init__(
self,
root: str,
train_test_val: str = None,
data_type: str = "event",
frames_number: int = None,
split_by: str = None,
duration: int = None,
custom_integrate_function: Callable = None,
custom_integrated_frames_dir_name: str = None,
transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None,
):
r"""
**API Language:**
:ref:`中文 <HARDVS.__init__-cn>` | :ref:`English <HARDVS.__init__-en>`
----
.. _HARDVS.__init__-cn:
* **中文**
HARDVS 数据集，由 `HARDVS: Revisiting Human Activity Recognition with Dynamic Vision Sensors. <https://arxiv.org/pdf/2211.09648>`_ 提出。
有关参数的更多详细信息，请参考 :class:`NeuromorphicDatasetFolder <spikingjelly.datasets.base.NeuromorphicDatasetFolder>`
----
.. _HARDVS.__init__-en:
* **English**
The HARDVS dataset, which is proposed by
`HARDVS: Revisiting Human Activity Recognition with Dynamic Vision Sensors. <https://arxiv.org/pdf/2211.09648>`_.
Refer to :class:`NeuromorphicDatasetFolder <spikingjelly.datasets.base.NeuromorphicDatasetFolder>` for more details about params information.
:param root: 数据集的根路径
:type root: Union[str, Path]
:param train_test_val: ``"train"``、``"val"`` 或 ``"test"``
:type train_test_val: str
:param data_type: ``"event"`` 或 ``"frame"``
:type data_type: str
:param frames_number: 积分帧的数量
:type frames_number: Optional[int]
:param split_by: ``"time"`` 或 ``"number"``
:type split_by: Optional[str]
:param duration: 每帧的时间时长
:type duration: Optional[int]
:param custom_integrate_function: 用户自定义积分函数
:type custom_integrate_function: Optional[Callable]
:param custom_integrated_frames_dir_name: 自定义积分帧目录名
:type custom_integrated_frames_dir_name: Optional[str]
:param transform: 数据变换
:type transform: Optional[Callable]
:param target_transform: 标签变换
:type target_transform: Optional[Callable]
:param root: Root directory of the dataset
:type root: Union[str, Path]
:param train_test_val: ``"train"``, ``"val"``, or ``"test"``
:type train_test_val: str
:param data_type: ``"event"`` or ``"frame"``
:type data_type: str
:param frames_number: Number of frames to integrate
:type frames_number: Optional[int]
:param split_by: ``"time"`` or ``"number"``
:type split_by: Optional[str]
:param duration: Time duration per frame
:type duration: Optional[int]
:param custom_integrate_function: User-defined integrate function
:type custom_integrate_function: Optional[Callable]
:param custom_integrated_frames_dir_name: Custom frames directory name
:type custom_integrated_frames_dir_name: Optional[str]
:param transform: Transform function
:type transform: Optional[Callable]
:param target_transform: Target transform function
:type target_transform: Optional[Callable]
:return: None
:rtype: None
"""
self.train_test_val = train_test_val
super().__init__(
root,
None,
data_type,
frames_number,
split_by,
duration,
custom_integrate_function,
custom_integrated_frames_dir_name,
transform,
target_transform,
)
[文档]
def get_root_when_train_is_none(self, _root: Path):
return _root / self.train_test_val
[文档]
@classmethod
def get_H_W(cls) -> Tuple:
"""
:return: ``(260, 346)``
"""
return 260, 346
[文档]
@classmethod
def resource_url_md5(cls) -> list:
url = "https://github.com/Event-AHU/HARDVS"
return [
("MINI_HARDVS_files.zip", url, "9c4cc0d9ba043faa17f6f1a9e9aff982"),
("test_label.txt", url, "5b664af5843f9b476a9c22626f7f5a59"),
("train_label.txt", url, "0d642b6e6871034f151b2649a89d8d3c"),
("val_label.txt", url, "cd2cebcba80e4552102bbacf2b5df812"),
]
[文档]
@classmethod
def downloadable(cls) -> bool:
"""
:return: ``False``
"""
return False
