Skip to content

lorenzinigiovanni/dl-project

Repository files navigation

Acknowledgment

This is a project for the course: Deep Learning of the University of Trento.

Professors:

  • Elisa Ricci
  • Willi Menapace

Students:

  • Giovanni Lorenzini 223715
  • Simone Luchetta 223716
  • Diego Planchenstainer 223728

Necessary packages

Using pip install the following packages.

$ pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
$ pip3 install pandas
$ pip3 install tensorboardX
$ pip3 install numpy

Dataset

The dataset needs to be extracted in a folder called dataset, mantaining the internal structure unchanged.

Code execution

Before execution is necessary to set to true or false the variable is_training in the main.py. If setted to true the network will train, this takes some time. If setted to false, and providing a network model (network.pt) is possible to go directly to the test mode. It will compute the mAP and the two required files (classification_test.csv and reid_test.txt). Then it's possible to execute the code simply by calling python and the path of the main file.

$ python3 main.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages