Object detection transfer learning with PyTorch. In this project, i've used transfer learning on efficentNetB0 model for bounding box regression. The project includes a jupyter notebook linked to google collab and python modules folder with the data/ model pre/post proccessing. The model was trained on a GPU.
PyTorch, Python, Jupyter notebook, transfer learning, efficentNetB0 model