Return results in sahi/predict.py

[imandrealombardo]

Hi,

I noticed that the function predict doesn't return results like in get_sliced_prediction. There, it returns object_prediction_list, a list of sahi.prediction.ObjectPrediction.

Would it be possible to have the predict function return a list of object predictions for each image in the batch?

[fcakyon]

@imandrealombardo you can set export_pickle=True

sahi/sahi/predict.py

Line 369 in 1240bab

export_pickle: bool = False,

and set return_dict=True

sahi/sahi/predict.py

Line 381 in 1240bab

return_dict: bool = False,

then predict will return {'export_dir': str} then you can load the pickle in export dir to access object_prediction_list 👍🏻