README.md

# Configure environment
## 1. Install Conda packages:

* Pytorch 1.8.1 with CUDA

```
$ conda install pytorch torchvision torchaudio cudatoolkit=<your cuda version> -c pytorch -c nvidia
```

Or ref to https://pytorch.org/get-started/locally/ for install guide

* matplotlib

* tensorboard

## 2. Install Pip packages:
* pyGlm
* tensorboardX
* (optional) opencv-python

# Useful commands
## 1. Video generate:
```
$ ffmpeg -y -r 30 -i view_%04d.png -c:v libx264 -vf fps=30 -pix_fmt yuv420p ../train.mp4
```

## 2. Convert onnx to tensorRT
```
$ trtexec --onnx=net@256x256x2.onnx --fp16 --saveEngine=net@256x256x2.trt --workspace=4096
```