SoccerRobotPerception- [Paper]
Detection of Goalpost, Ball, Robot and Segmentation of field and fieldlines with a two headed network.
- PyTorch
- Numpy
- Matplotlib
- OpenCV
- Scikit-Image
To train the model, please run
$ python train.py --lr=<learning_rate> --epochs=<epochs> --blob_dir=<Detection images path> --seg_dir=<Segmentation images path>
An additional visualizations.ipynb is added to perform the testing and visualize the results. Please run this file to see the visualizations
Our sincere thanks for Hafez Farazi from University of Bonn for his continous support and advice during this project.