stuff i did but forgot

gustavz · Apr 18, 2018 · 23708c8 · 23708c8
1 parent 8d7c053
commit 23708c8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/object_detection.py b/object_detection.py
@@ -62,7 +62,7 @@ def download_model():
         tar_file = tarfile.open(model_file)
         for file in tar_file.getmembers():
           file_name = os.path.basename(file.name)
-          if 'toy_frozen_inference_graph.pb' in file_name:
+          if 'frozen_inference_graph.pb' in file_name:
             tar_file.extract(file, os.getcwd() + '/models/')
         os.remove(os.getcwd() + '/' + model_file)
     else:

diff --git a/object_detection_kcf_test.py b/object_detection_kcf_test.py
@@ -69,7 +69,7 @@ def download_model():
         tar_file = tarfile.open(model_file)
         for file in tar_file.getmembers():
           file_name = os.path.basename(file.name)
-          if 'toy_frozen_inference_graph.pb' in file_name:
+          if 'frozen_inference_graph.pb' in file_name:
             tar_file.extract(file, os.getcwd() + '/models/')
         os.remove(os.getcwd() + '/' + model_file)
     else: