Linkit WS22-23- Intermediate Challenge
- Install Dependencies
pip install -r requirements.txt
Note: Requires Python 3.6-3.10
- Open challenge.ipynb
.
├── README.md
├── challenge.ipynb
├── datasets
│ ├── yourdatasetname
│ │ ├── images
│ │ │ ├── train
│ │ │ │ ├── 0.jpg
│ │ │ │ ├── 1.jpg
│ │ │ ├── val
│ │ │ ├── test-dev
│ │ ├── labels
│ │ │ ├── train
│ │ │ │ ├── 0.txt
│ │ │ │ ├── 1.txt
│ │ │ ├── val
│ │ │ ├── test-dev
│── .gitignore
├── requirements.txt
├── yolov5 [GIT SUBMODULE]
It is important to monitor the training process to ensure that the model is training properly. To do so, we recommend Weights and Biases (or tensorboard). Both tools keep track of the training process and automatically log the results.
- Create an account on Weights and Biases
- Install the wandb package
pip install wandb
- Login to your account
wandb login
- Run the training script with the
--project
flagpython train.py --project <project_name>
- Go to your Weights and Biases dashboard to view the results
For more information on the YOLOV5 integration with Weights and Biases, refer to here
If installed, training will automatically log to tensorboard.
-
Fill in the
inference
function in challenge.ipynb- Inference expects a model
- Note: Feel free to borrow code from the yolov5 detect.py
-
Update your requirements.txt file if you have installed any additional packages:
pip list --format=freeze > requirements.txt
-
We will run the code with github actions using the following command:
python submission.py
-
The Output of the code should be a pandas dataframe file in the same format as the exemplary
inference
functions output -
You can check your current fps via the actions tab on github
- Open the actions tab
- Click on your latest action
- Open the
Run Challenge
step and scroll down to theTest Results
section
We will evaluate your model on the following metrics:
- Mean Average Precision (mAP)
- FPS (Frames per second) must be greater than 5
- Model Size
BONUS: Can you convert and run your model in the onnx format? This may increase inference speed and reduce model size.
If you have any questions, please contact us on linkit-Teams (or email us) at
- Philipp Kiesling: [email protected]
- Justus Thomsen: [email protected]