SforAiDl / Playground

A python library consisting of pipelines for visual analysis of different sports using Computer Vision and Deep Learning.
MIT License
18 stars 17 forks source link

Created the save_prediction option #29

Closed ShreyPandit closed 4 years ago

ShreyPandit commented 4 years ago

created the save prediction option inside the badminton.py

Sharanranjit commented 4 years ago

Hey! Just a correction, the input image is not always a path, it can be a loaded array also. So you would need to check for that too. And just a suggestion : Try saving it using opencv, by making a copy of original image(it is already done as out_img)