Commit 98384eb
Wildlife: Error NMS (#212)
* Reduce batch size from 16 to 8 for detection
I have an issue with the Wildlife tool. It works correctly on my computer when running on CPU, but in Galaxy with GPU I get the error: "NMS time limit 0.850s exceeded".
As I understand it, the problem is that the GPU is faster and tries to process more images, which increases the processing time of the NMS step. To handle this, the system removes or skips some detections to avoid exceeding the time limit.
Therefore, one possible solution would be to reduce the "batch_size".
* Add use_fast option to image classification model
* Update wildlife_detection_classification.xml
* Clarify description with AI reference
Added '(AI)' to the description for clarification.
* Fix indentation in model loading section
* Enable fast processing in image processor initialization
* Fix formatting
* Add batch_size parameter to wildlife detection
* Add batch_size argument for image processing
* formatting
* use safetensors as datatype and small secutiry changes
* Implement time limit NMS
* Update wildlife detection tool version and parameters
* flake8
* Remove one bad use_fast
* Flake8
Refactored imports and removed unused code.
* Flake8.2
* Flake8.3
* Flake8.4
* Flake8.5
* Update Wildlife Detection tool version to 0.1.1
---------
Co-authored-by: Björn Grüning <bjoern@gruenings.eu>1 parent d329c54 commit 98384eb
2 files changed
Lines changed: 280 additions & 63 deletions
0 commit comments