mirror of
https://github.com/openmv/openmv.git
synced 2025-11-04 14:49:50 +08:00
scripts/libraries: Load model's labels from file.
Signed-off-by: iabdalkader <i.abdalkader@gmail.com>
This commit is contained in:
parent
5f369df045
commit
3f55d956c5
@ -34,6 +34,11 @@ from ml.preprocessing import Normalization
|
|||||||
class Model(uml.Model):
|
class Model(uml.Model):
|
||||||
def __init__(self, *args, **kwargs):
|
def __init__(self, *args, **kwargs):
|
||||||
super().__init__(*args, kwargs.get("load_to_fb", False))
|
super().__init__(*args, kwargs.get("load_to_fb", False))
|
||||||
|
try:
|
||||||
|
path = args[0].split(".")[0] + ".txt"
|
||||||
|
self.labels = [line.rstrip('\n') for line in open(path, "r")]
|
||||||
|
except Exception:
|
||||||
|
self.labels = None
|
||||||
|
|
||||||
def predict(self, args, **kwargs):
|
def predict(self, args, **kwargs):
|
||||||
args = [Normalization()(x) if isinstance(x, image.Image) else x for x in args]
|
args = [Normalization()(x) if isinstance(x, image.Image) else x for x in args]
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user