Skip to content
Snippets Groups Projects
Commit c66b09a1 authored by Stephane Chavin's avatar Stephane Chavin
Browse files

Update file get_train_annot.py

parent 854c810c
Branches
No related tags found
No related merge requests found
......@@ -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]),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment