diff --git a/get_train_annot.py b/get_train_annot.py index 731cc9d2da61f7dde115cdc47c3c08a67d7f8c28..a4ea16e1c60413355600084c41e9112100b8017b 100755 --- a/get_train_annot.py +++ b/get_train_annot.py @@ -83,8 +83,8 @@ def main(entry, arguments, species_list): (arguments.rf / 2) # take height value in pixels # Correction if the boxes are corrupted (> 1 or < 0) - x_pxl, width_pxl = correct_box(x_pxl, width_pxl) - y_pxl, height_pxl = correct_box(y_pxl, height_pxl) + x_pxl, width_pxl = utils.correct_box(x_pxl, width_pxl) + y_pxl, height_pxl = utils.correct_box(y_pxl, height_pxl) # Store the annotation in a DataFrame new_table = pd.DataFrame([[str(species_list[species_list.species == specie].index[0]),