Marmelator/image-classification-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

removed redundant TF env var export

Browse Source
This commit is contained in:
Matthias Bisping 2022-04-01 21:35:10 +02:00
parent c80549d5d3
commit 3d1c251e10
3 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
image_prediction/redai_adapter/efficient_net_wrapper.py
View File

@ -1,11 +1,7 @@
import os import tensorflow as tf
from image_prediction.redai_adapter.model_wrapper import ModelWrapper from image_prediction.redai_adapter.model_wrapper import ModelWrapper
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import tensorflow as tf
class EfficientNetWrapper(ModelWrapper): class EfficientNetWrapper(ModelWrapper):

1
image_prediction/redai_adapter/mlflow.py
View File

@ -31,7 +31,6 @@ class MlflowModelReader:
@lru_cache(maxsize=None) @lru_cache(maxsize=None)
def __get_run(self, run_id): def __get_run(self, run_id):
return mlflow.get_run(run_id) return mlflow.get_run(run_id)
def __get_classes(self, run_id, prefix="tt"): def __get_classes(self, run_id, prefix="tt"):

4
image_prediction/redai_adapter/model_wrapper.py
View File

@ -1,10 +1,6 @@
import abc import abc
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf