兼容tf2+

kerlomz · Nov 21, 2020 · e568d39 · e568d39
1 parent f2f0a31
commit e568d39
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/make_dataset.py b/make_dataset.py
@@ -68,7 +68,7 @@ def convert_dataset_from_filename(self, output_filename, file_list, mode: RunMod
                     if labels:
                         labels = labels.group()
                     else:
-                        tf.logging.warning('invalid filename {}, ignored.'.format(file_name))
+                        tf.compat.v1.logging.warning('invalid filename {}, ignored.'.format(file_name))
                         continue
                         # raise NameError('invalid filename {}'.format(file_name))
                     labels = labels.encode('utf-8')

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
 pillow>=4.3.0
 opencv-python
-numpy==1.18.4
+numpy==1.18.5
 pyyaml>=3.13
 tqdm
 colorama