Fixed import; added new req item

AICAN-Research · Apr 29, 2024 · a39169e · a39169e
1 parent 68c4a61
commit a39169e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/SegmentationTraining.py b/SegmentationTraining.py
@@ -6,7 +6,7 @@
 import tensorflow as tf
 import tf2onnx
 from tensorflow.keras.models import load_model
-from MLD import multi_lens_distortion
+from src.augmentation.MLD import multi_lens_distortion
 from tensorflow.keras import Input
 from tensorflow.keras import layers
 from tensorflow.keras import models

diff --git a/requirements.txt b/requirements.txt
@@ -5,4 +5,5 @@ opencv-python
 matplotlib
 scikit-learn
 tf2onnx
-onnx
+onnx
+Pillow