Hugging Face's logo

Spaces:
rerun
/
Vista
Build error

App Files Community
1
Vista
File size: 1,667 Bytes 
d323598
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
## Installation

- ### Requirement

  Our experiments are conducted with **PyTorch 2.0.1**, **CUDA 11.7**, **Ubuntu 22.04**, and **NVIDIA Tesla A100** (80 GB). For other requirements, please check [TRAINING.md](https://github.com/OpenDriveLab/Vista/blob/main/docs/TRAINING.md) and [SAMPLING.md](https://github.com/OpenDriveLab/Vista/blob/main/docs/SAMPLING.md).

- ### Preparation

  Clone the repository to your local directory.

  ```shell
  git clone https://github.com/OpenDriveLab/Vista.git
  ```

  We provide an example on nuScenes dataset for training and sampling. Before you start, make sure you have:
  
  - Downloaded the translated action annotations from [here](https://drive.google.com/drive/folders/1JpZObdR0OXagCbnPZfMSI8vhGLom5pht?usp=sharing) and put the JSON files into `annos`.
    
  - Downloaded all splits of **Trainval** in **Full dataset (v1.0)** to your device following [official instructions](https://www.nuscenes.org/download). After downloading, it should contain:

    ```
    $<your-nusc-data-root>
    ├── samples
    ├── sweeps
    ├── ...
    └── v1.0-trainval
    ```

- ### Installation

  - We use conda to manage the environment.

    ```shell
    conda create -n vista python=3.9 -y
    conda activate vista
    ```
  
  - Install dependencies.
  
    ```shell
    conda install -y pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia
    pip3 install -r requirements.txt
    pip3 install -e git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata
    ```

---

=> Next: [[Training](https://github.com/OpenDriveLab/Vista/blob/main/docs/TRAINING.md)]