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(