diff --git a/get_train_annot.py b/get_train_annot.py index 6d680feebdd08643e2915e05bb962799b570ee1e..4b2bd2c4f5e4d96c1c92a831650b88e3205e7def 100755 --- a/get_train_annot.py +++ b/get_train_annot.py @@ -84,7 +84,8 @@ def main(entry, arguments, species_list): # Correction if the boxes are corrupted (> 1 or < 0) x_pxl, width_pxl = utils.correct_box(x_pxl, width_pxl) y_pxl, height_pxl = utils.correct_box(y_pxl, height_pxl) - + if x_pxl > 1 and width_pxl < 0.05: + continue # Store the annotation in a DataFrame new_table = pd.DataFrame([[str(species_list[species_list.species == specie].index[0]), x_pxl, y_pxl, width_pxl, height_pxl]],