Merge branch 'main' of https://gitlab.lis-lab.fr/stephane.chavin/psi-biom

README.md

@@ -33,7 +33,7 @@ python3 train.py --img IMG_SIZE --batch BATCH_SIZE --EPOCHS NB_EPOCHS --data DIR
 
 * Sauvegarde les annotations en .txt ainsi que les images avec les bounding box dessus
 ```bash
-python3 detect.py --weights ../psi-biom/best.pt --img 224 --conf 0.X --source PATH_TO_SPECTROGRAM_TO_DETECT --save-txt
+python3 detect.py --weights ../psi-biom/best.pt --img 224 --conf 0.X --source PATH_TO_IMG --save-txt
 ```
 
 **WARNING** : Il faut adapter EXP_NB, qui correspond au numéro de l'entrainement *(exp1 pour le premier entrainement)* 
@@ -42,7 +42,7 @@ python3 detect.py --weights ../psi-biom/best.pt --img 224 --conf 0.X --source PA
 * Sauvegarde les annotations en .txt seulement avec la confiance de chaque détections
 
 ```bash
-python3 detect.py --weights ..psi-biom/best.pt --img 224 --conf 0.X --source PATH_TO_SPECTROGRAM_TO_DETECT --save-txt --nosave --save-conf
+python3 detect.py --weights ..psi-biom/best.pt --img 224 --conf 0.X --source PATH_TO_IMG --save-txt --nosave --save-conf
 ``` 
 </details>
 
@@ -53,7 +53,7 @@ python3 detect.py --weights ..psi-biom/best.pt --img 224 --conf 0.X --source PAT
 
 ```bash
 cd ../
-python3 get_yolo_detection.py -p PATH_TO_YOUR_IMG -d DIRECTION_OF_THE_RESULT
+python3 get_yolo_detection.py -p PATH_TO_THE_TXT -d DIRECTION_OF_THE_RESULT #PATH_TO_THE_TXT ~ yolov5/runs/detect/expX/labels/
 ```
 **WARNING** : Il est important d'ajouter sa propre liste de classes (ligne 79)