Live demo -> https://huggingface.co/spaces/loeclos/aircraft-prediction
0.104/07/2025- Created repository.
| resources # Folder with different files used for showing the model in action.
│ model_weights.pt # The best model weights.
│ main.ipynb # The notebook that the model was trained with
│ classes.csv # The available classes.
This project aims to be able to classify planes from the FGCV dataset. The model we use here is the yolo11n from the ultralitycs library. The dataset from pytorch datasets.
Everything else is quite easy. Half of the notebook is focused on preprocessing the data. After that we load our yolo model using the weights in the repo.
You can see the model in action here: https://colab.research.google.com/drive/18nlsrKk2FIlxxOU16HT0RSUw-LVUwgGy?usp=sharing