# IGNet 

## Environment

Please run `conda create -f enviomnent.yaml` for installing the environment for training and evaluating IGNet. 

## Training

Run `generate_split.py` for generating split for training and validation dataset. 

Run `python train.py --config configs/ignet_goal.yaml` for training IGNet. The visualization and logs will be shown in weights and biases. 

## Evaluation

Run `python eval.py --config configs/eval_ignet.yaml` for evaluating IGNet. The results will be shown in `videos` directory. 

A video generated by trained IGNet by us is attached in `videos/`.  