|
This directory contains the `object_detection.py` module, which provides functionality for detecting objects in images using various models. |
|
The `object_detection.py` module is designed to detect objects in images using object detection models like Detic and YOLOv5. |
|
It includes methods for loading models, processing images, detecting objects, and drawing bounding boxes around detected objects. |
|
|
|
Notes |
|
Ensure the necessary dependencies are installed (streamlit, torch, PIL, transformers, numpy, cv2, os). |
|
The module supports processing images from file paths or file-like objects. |
|
The object detection functionality relies on the Detic and YOLOv5 models. Make sure to have these models properly configured and accessible from `models` directory. |