Skip to content

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

License

Notifications You must be signed in to change notification settings

DRL-CASIA/NeuronsDataset

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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 [email protected]: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.

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages