search

ValBaron10 / BirdSoundClassif

2 stars 0 forks source link

Feature/dockerization #9

Closed MatthieuLeNozach closed 1 month ago

MatthieuLeNozach commented 1 month ago

La classification tourne correctement chez moi

Modifs pour la dockerization: cf https://github.com/ValBaron10/BirdSoundClassif/blob/feature/dockerization/README_DOCKER.md

Modifs nécessaires pour utiliser le cpu:

Ajout d'une fonction load_weights_cpu() dans src/models/detr Ajout d'un script spécifique au cpu run_detection_cpu.py dans src/models

Résultat:

root@0a0df020b2e4:/app# python inference/main.py 
...
2024-05-08 10:28:03,038 - INFO - [output]: 
{'Turdus merula': {'bbox_coord': [[552, 182, 629, 258]], 'scores': [0.9958606362342834]}}