Closed MatthieuLeNozach closed 4 months 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:
load_weights_cpu()
src/models/detr
run_detection_cpu.py
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]}}
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:
load_weights_cpu()
danssrc/models/detr
run_detection_cpu.py
danssrc/models
Résultat: