Fix #143

.gitignore

@@ -20,4 +20,5 @@ road_sign_data.yaml
 BCCD_Dataset/
 model_training/
 __pycache__
-samples
+samples
+build/

pylabel/analyze.py

@@ -135,7 +135,7 @@ def move_column_inplace(df, col, pos):
             df.insert(pos, col.name, col)
 
         df_value_counts = pd.DataFrame(
-            ds.df["cat_name"].value_counts(normalize=normalize), columns=["cat_name"]
+            ds.df["cat_name"].value_counts(normalize=normalize)
         )
 
         df_value_counts.index.name = "cat_name"
@@ -150,12 +150,13 @@ def move_column_inplace(df, col, pos):
             group_df = pd.DataFrame(group)
             df_split_value_counts = pd.DataFrame(
                 group_df["cat_name"].value_counts(normalize=normalize),
-                columns=["cat_name"],
             )
             df_split_value_counts.index.name = "cat_name"
             df_split_value_counts.columns = [name]
             df_value_counts = pd.merge(
-                df_value_counts, df_split_value_counts, how="left", on=["cat_name"]
+                df_value_counts, df_split_value_counts, 
+                how="left", 
+                left_index=True, right_index=True
             )
 
         # Move 'train' to the left of the table since that is the usual convention.

pylabel/importer.py

@@ -300,6 +300,8 @@ def ImportYoloV5(
 
     def GetCatNameFromId(cat_id, cat_names):
         cat_id = int(cat_id)
+        if not cat_names:
+            return str(cat_id)
         if len(cat_names) > int(cat_id):
             return cat_names[cat_id]