moved callbacks to dataset_iterator

jeanollion · May 22, 2024 · 8373de0 · 8373de0
1 parent 059218f
commit 8373de0
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 146 deletions.
diff --git a/distnet_2d/utils/__init__.py b/distnet_2d/utils/__init__.py
@@ -1,4 +1,3 @@
 name="utils"
-from .callbacks import StopOnLR, EpsilonCosineDecayCallback, LogsCallback, SafeModelCheckpoint, ReduceLROnPlateau2
 from .helpers import predict_average_flip_rotate
 from .losses import PseudoHuber, weighted_loss_by_category, balanced_category_loss
diff --git a/distnet_2d/utils/callbacks.py b/distnet_2d/utils/callbacks.py
diff --git a/setup.py b/setup.py
@@ -5,14 +5,14 @@
 
 setuptools.setup(
     name="DiSTNet2D",
-    version="0.1.6",
+    version="0.1.7",
     author="Jean Ollion",
     author_email="jean.ollion@polytechnique.org",
     description="tensorflow/keras implementation of DiSTNet 2D",
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/jeanollion/distnet2d",
-    download_url='https://github.com/jeanollion/distnet2d/releases/download/v0.1.6/distnet2d-0.1.6.tar.gz',
+    download_url='https://github.com/jeanollion/distnet2d/releases/download/v0.1.7/distnet2d-0.1.7.tar.gz',
     packages=setuptools.find_packages(),
     keywords=['Segmentation', 'Tracking', 'Cell', 'Tensorflow', 'Keras'],
     classifiers=[
@@ -24,5 +24,5 @@
         'Programming Language :: Python :: 3',
     ],
     python_requires='>=3',
-    install_requires=['numpy', 'scipy', 'tensorflow>=2.7.1', 'edt>=2.0.2', 'scikit-fmm', 'numba', 'dataset_iterator>=0.4.4', 'elasticdeform>=0.4.7']
+    install_requires=['numpy', 'scipy', 'tensorflow>=2.7.1', 'edt>=2.0.2', 'scikit-fmm', 'numba', 'dataset_iterator>=0.4.5', 'elasticdeform>=0.4.7']
 )