refactorize slicing (#95)

* refactorize slicing * update dev dependencies * minor refactorization * update notebooks * remove unnesessary imports * add pil image reading * add pillow to reqs * refactorize slicing * update tests
obss · May 15, 2021 · 31bc7e1 · 31bc7e1
1 parent 3058da1
commit 31bc7e1
Show file tree

Hide file tree

Showing 11 changed files with 564 additions and 438 deletions.
diff --git a/demo/inference.ipynb b/demo/inference.ipynb
diff --git a/demo/slicing.ipynb b/demo/slicing.ipynb
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 opencv-python>=4.2.0.32
 shapely>=1.7.0
 tqdm>=4.48.2
-dataclasses; python_version<"3.7"
+dataclasses; python_version<"3.7"
+pillow>=8.2.0
diff --git a/sahi/model.py b/sahi/model.py
@@ -1,12 +1,9 @@
 # OBSS SAHI Tool
 # Code written by Fatih C Akyon, 2020.
 
-import os
-import sys
-
 import numpy as np
 
-from sahi.prediction import ObjectPrediction, PredictionInput
+from sahi.prediction import ObjectPrediction
 from sahi.utils.torch import cuda_is_available, empty_cuda_cache
 
 

diff --git a/sahi/predict.py b/sahi/predict.py
@@ -19,7 +19,6 @@
 )
 from sahi.utils.file import (
     Path,
-    get_base_filename,
     import_class,
     increment_path,
     list_files,
@@ -172,7 +171,7 @@ def get_sliced_prediction(
 
     # create slices from full image
     time_start = time.time()
-    slice_image_result, num_total_invalid_segmentation = slice_image(
+    slice_image_result = slice_image(
         image=image,
         slice_height=slice_height,
         slice_width=slice_width,