DevoLearn / devolearn

Accelerate data driven research on embryos with deep learning models :microscope:
MIT License
43 stars 30 forks source link

Enhancement: Allow prediction on numpy arrays present as local variable #47

Open vrutikrabadia opened 3 years ago

vrutikrabadia commented 3 years ago

The current model only allows us to predict on images by providing a path. We can extend the functionality and flexibility of out module by allowing numpy arrays as arguments to predict.

Mayukhdeb commented 3 years ago

You're right, I'm planning to add all of these functionalities through the InferenceEngine class found here

So a more sensible thing to do now is to modify all of the current models so that they're a subclass of the InferenceEngine class.

vrutikrabadia commented 3 years ago

Okay

vrutikrabadia commented 3 years ago

So first we can do that and then continue with this.

Mayukhdeb commented 3 years ago

yes