vedatad

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

(★ 55)

A single stage temporal action detection toolbox based on PyTorch

vedatad Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

## Introduction
Vedatad is a single stage temporal action detection toolbox based on PyTorch. Vedatad includes implementations of the following temporal action detection algorithms:

* [DaoTAD](https://github.com/Media-Smart/vedatad/tree/main/configs/trainval/daotad)

## Features

- **Modular Design**

We decompose detector into four parts: data pipeline, model, postprocessing and criterion which make it easy to convert PyTorch model into TensorRT engine and deploy it on NVIDIA devices such as Tesla V100, Jetson Nano and Jetson AGX Xavier, etc.

- **Support of several popular single stage detector**

The toolbox supports several single stage detector out of the box, *e.g.* DaoTAD, etc.

## License

This project is released under the [Apache 2.0 license](LICENSE).

## Installation
### Requirements

- Linux
- Python 3.7+
- PyTorch 1.7.0 or higher
- CUDA 10.2 or higher
- ffmpeg

We have tested the following versions of OS and softwares:

- OS: Ubuntu 16.04.6 LTS
- CUDA: 10.2
- PyTorch 1.8.0
- Python 3.8.5
- ffmpeg 4.3.11

### Install vedatad

a. Create a conda virtual environment and activate it.

```shell
conda create -n vedatad python=3.8.5 -y
conda activate vedatad
```

b. Install PyTorch and torchvision following the [official instructions](https://pytorch.org/), *e.g.*,

```shell
conda install pytorch torchvision -c pytorch
```

c. Clone the vedatad repository.

```shell
git clone https://github.com/Media-Smart/vedatad.git
cd vedatad
vedatad_root=${PWD}
```

d. Install vedatad.

```shell
pip install -r requirements/build.txt
pip install -v -e .
```

## Data preparation

Please follow specified algorithm in `config/trainval` to prepare data, for example, see detail in `configs/trainval/daotad`.

## Train

a. Config

Modify some configuration accordingly in the config file like `configs/trainval/daotad/daotad_i3d_r50_e700_thumos14_rgb.py`

b. Train
```shell
tools/dist_trainval.sh configs/trainval/daotad/daotad_i3d_r50_e700_thumos14_rgb.py "0,1,2,3"
```

## Test

a. Config

Modify some configuration accordingly in the config file like `configs/trainval/daotad/daotad_i3d_r50_e700_thumos14_rgb.py`

b. Test
```shell
CUDA_VISIBLE_DEVICES=0 python tools/test.py configs/trainval/daotad/daotad_i3d_r50_e700_thumos14_rgb.py weight_path
```

## Contact

This repository is currently maintained by Hongxiang Cai ([@hxcai](http://github.com/hxcai)), Yichao Xiong ([@mileistone](https://github.com/mileistone)), Chenhao Wang ([@C-H-Wong](https://github.com/C-H-Wong)).

## Credits
We got a lot of code from [vedadet](https://github.com/Media-Smart/vedadet), thanks to [Media-Smart](https://github.com/Media-Smart).

vedatad

vedatad Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

vedatad

vedatad Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README