We borrow the codes and implementations from jwyang-faster-rcnn.pytorch, and the Pytorch version is 1.0. Please refer to jwyang-faster-rcnn.pytorch for more details of setups and implementations.
Below is the framework of proposed method:
The dataset link is golf_ball.
Some tracking results are shown below:
@article{zhang2020efficient,
title={Efficient Golf Ball Detection and Tracking Based on Convolutional Neural Networks and Kalman Filter},
author={Zhang, Tianxiao and Zhang, Xiaohan and Yang, Yiju and Wang, Zongbo and Wang, Guanghui},
journal={arXiv preprint arXiv:2012.09393},
year={2020}
}
@inproceedings{zhang2020real,
title={Real-time golf ball detection and tracking based on convolutional neural networks},
author={Zhang, Xiaohan and Zhang, Tianxiao and Yang, Yiju and Wang, Zongbo and Wang, Guanghui},
booktitle={2020 IEEE International Conference on Systems, Man, and Cybernetics (SMC)},
pages={2808--2813},
year={2020},
organization={IEEE}
}