/NeuronsDataset

The official dataloader and visualization tools for RoboMaster team CASIA-Neurons

Primary LanguagePythonMIT LicenseMIT

Neurons Dataset API - The official dataloader and visualization tools for Neurons Datasets.

Introduction

We propose our dataloader API for loading and visualizing our RoboMaster opensource datasets. The datasets includes:

Dependencies

Note that this API is only tested on ubuntu/Mac devices which has:

  • python 3.6+
  • torch 1.7.1
  • torchvision 0.8.2
  • ...

Installation

First, clone this repo:

git clone git@github.com:DRL-CASIA/NeuronsDataset.git
cd NeuronsDataset

Then, pack the code into wheel packages for later installation:

conda activate YourEnv  # Activate your python environment first.
cd src
pip3 setup.py bdist_wheel

The commands above are expected to generate a whl file and several folders.

Next, install the whl package into your python3 environment using pip:

cd dist
pip3 install neuronsdataset-1.0-py3-none-any.whl

The generated folders could be safely deleted after installing the API.

Usage

Please refer to main.py for details.

We also provide samples of dataset mentioned in Introduction section, which are contaied in sample_data.