VRDL_project1

Requirements

To install requirements:

pip install -r requirements.txt

Hardware

NVIDIA GeForce 3090

Training

To train the model, run this command:

python main.py

Evaluation

To evaluate my model, run:

python inference.py

Model Weight Link

https://drive.google.com/drive/folders/1qMi1xaPWkpJr7vNadyh4fJ_niqN22O92?usp=sharing