Changes to class maps

hwanji · Jan 2, 2020 · 097d33d · 097d33d
1 parent ec0dd4c
commit 097d33d
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 5 deletions.
diff --git a/detector/taco_config/map_1.csv b/detector/taco_config/map_1.csv
@@ -37,7 +37,7 @@ Polypropylene bag,Litter
 Produce bag,Litter
 Cereal bag,Litter
 Bread bag,Litter
-Plastic Film,Litter
+Plastic film,Litter
 Crisp packet,Litter
 Other plastic wrapper,Litter
 Retort pouch,Litter

diff --git a/detector/taco_config/map_17.csv b/detector/taco_config/map_17.csv
@@ -37,7 +37,7 @@ Polypropylene bag,Plastic film
 Produce bag,Plastic film
 Cereal bag,Plastic film
 Bread bag,Plastic film
-Plastic Film,Plastic film
+Plastic film,Plastic film
 Crisp packet,Wrapper
 Other plastic wrapper,Wrapper
 Retort pouch,Wrapper

diff --git a/detector/taco_config/map_2.csv b/detector/taco_config/map_2.csv
@@ -37,7 +37,7 @@ Polypropylene bag,Background
 Produce bag,Background
 Cereal bag,Background
 Bread bag,Background
-Plastic Film,Background
+Plastic film,Background
 Crisp packet,Background
 Other plastic wrapper,Background
 Retort pouch,Background

diff --git a/detector/taco_config/map_3.csv b/detector/taco_config/map_3.csv
@@ -37,7 +37,7 @@ Polypropylene bag,Plastic bag + wrapper
 Produce bag,Plastic bag + wrapper
 Cereal bag,Plastic bag + wrapper
 Bread bag,Plastic bag + wrapper
-Plastic Film,Plastic bag + wrapper
+Plastic film,Plastic bag + wrapper
 Crisp packet,Plastic bag + wrapper
 Other plastic wrapper,Plastic bag + wrapper
 Retort pouch,Plastic bag + wrapper

diff --git a/detector/taco_config/map_4.csv b/detector/taco_config/map_4.csv
@@ -37,7 +37,7 @@ Polypropylene bag,Plastic bag + wrapper
 Produce bag,Plastic bag + wrapper
 Cereal bag,Plastic bag + wrapper
 Bread bag,Plastic bag + wrapper
-Plastic Film,Plastic bag + wrapper
+Plastic film,Plastic bag + wrapper
 Crisp packet,Plastic bag + wrapper
 Other plastic wrapper,Plastic bag + wrapper
 Retort pouch,Plastic bag + wrapper

diff --git a/detector/utils.py b/detector/utils.py
@@ -881,6 +881,11 @@ def download_trained_weights(coco_model_path, verbose=1):
 
     coco_model_path: local path of COCO trained weights
     """
+
+    subdir = os.path.dirname(coco_model_path)
+    if not os.path.isdir(subdir):
+        os.mkdir(subdir)
+
     if verbose > 0:
         print("Downloading pretrained model to " + coco_model_path + " ...")
     with urllib.request.urlopen(COCO_MODEL_URL) as resp, open(coco_model_path, 'wb') as out: