Merge branch 'master' of git://github.com/laxos96/mAP into laxos96-ma…

…ster
jesusoctavioas · Jun 15, 2018 · a5fbc13 · a5fbc13
2 parents b94065e + 605581f
commit a5fbc13
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/extra/remove_space.py b/extra/remove_space.py
@@ -77,8 +77,8 @@ def rename_class(current_class_name, new_class_name):
 
 with open('class_list.txt') as f:
     for line in f:
-        current_class_name = line
-        new_class_name = line.replace(' ', args.delimiter)
+        current_class_name = line.rstrip("\n")
+        new_class_name = line.replace(' ', args.delimiter).rstrip("\n)
         if line == new_class_name:
             continue
         y_n_message = ("Are you sure you want "
@@ -94,4 +94,4 @@ def rename_class(current_class_name, new_class_name):
           os.chdir("../predicted")
           rename_class(current_class_name, new_class_name)
 
-print('Done!')
+print('Done!')