From e0942e826e4f5bc188422b26eb068da1af23ea34 Mon Sep 17 00:00:00 2001 From: brimoor Date: Wed, 12 Jan 2022 21:50:51 -0500 Subject: [PATCH] don't delete archives here --- eta/detectors/efficientdet.py | 2 +- eta/detectors/tfmodels_detectors.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/eta/detectors/efficientdet.py b/eta/detectors/efficientdet.py index 4d0b879e..03882669 100644 --- a/eta/detectors/efficientdet.py +++ b/eta/detectors/efficientdet.py @@ -111,7 +111,7 @@ def __init__(self, config): self._model_dir = etau.split_archive(model_path)[0] if not os.path.isdir(self._model_dir): logger.info("Extracting archive '%s'", model_path) - etau.extract_archive(model_path, delete_archive=True) + etau.extract_archive(model_path) # Load class labels self._labels_map = etal.load_labels_map(self.config.labels_path) diff --git a/eta/detectors/tfmodels_detectors.py b/eta/detectors/tfmodels_detectors.py index 5b3b26ed..2d4fdb22 100644 --- a/eta/detectors/tfmodels_detectors.py +++ b/eta/detectors/tfmodels_detectors.py @@ -447,7 +447,7 @@ def __init__(self, config): self._model_dir = etau.split_archive(model_path)[0] if not os.path.isdir(self._model_dir): logger.info("Extracting archive '%s'", model_path) - etau.extract_archive(model_path, delete_archive=True) + etau.extract_archive(model_path) # Load labels self._category_index, self._class_labels = _parse_labels_map(