COMET Phase-I Track Reconstruction using Machine Learning and Computer Vision