search

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

Save predictions as jpg #18

Closed ashwinvaswani closed 4 years ago

ashwinvaswani commented 4 years ago

Save figure flag in the detector has not been implemented yet. With an argument similar to display_detection, allow an argument for save figure or save prediction.

ShreyPandit commented 4 years ago

I would like to take up this issue. Kindly guide how to proceed.

ashwinvaswani commented 4 years ago

In detect_players_image method, add another argument which is False by default and when True, saves the prediction plot as jpg/png and prints a message.