Skip to content

Commit

Permalink
Merge pull request #144 from PUTvision/fix_error_after_installation
Browse files Browse the repository at this point in the history
Remove unnecessary import causing errors
  • Loading branch information
przemyslaw-aszkowski authored Feb 13, 2024
2 parents 0a8e8c9 + f31eb02 commit a6b78ef
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/deepness/processing/models/model_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import json
from typing import List, Optional

import cv2
import numpy as np

import deepness.processing.models.preprocessing_utils as preprocessing_utils
Expand Down Expand Up @@ -84,7 +83,7 @@ def get_model_batch_size(self) -> Optional[int]:
Batch size or None if not found (dynamic batch size)
"""
bs = self.input_shape[0]

if isinstance(bs, str):
return None
else:
Expand Down Expand Up @@ -183,7 +182,7 @@ def get_metadata_standarization_parameters(self) -> Optional[StandardizationPara
meta = self.sess.get_modelmeta()
name_mean = 'standardization_mean'
name_std = 'standardization_std'

param = StandardizationParameters(channels_number=self.get_input_shape()[-3])

if name_mean in meta.custom_metadata_map and name_std in meta.custom_metadata_map:
Expand Down

0 comments on commit a6b78ef

Please sign in to comment.