DeepStream-Yolo
DeepStream-Yolo copied to clipboard
Add support for YOLO-World
Hello, I appreciate your outstanding work on the DeepStream-YOLO repository, which has been incredibly helpful for using various YOLO models within the DeepStream SDK.
As the YOLO-World model, a real-time open-vocabulary object detector, has been recently released, I have extended your repository to include support for YOLO-World within the DeepStream SDK.
- [x] Added
utils/export_yoloWorld.py
- [x] Load YOLO-World model
- [x] Set custom classes (optional)
- [x] Modify outputs by adding a module,
DeepStreamOutput()
- [x] Export to ONNX
- [x] Added
config_infer_primary_yoloWorld.txt
- [x] Added
docs/YOLO_World.md
@marcoslucianops
Have you completed the TensorRT acceleration for YOLO-World? If so, that's really great work.
I appreciate it! With the DeepStream-YOLO by @marcoslucianops, I was able to make YOLO-World work within the DeepStream SDK. The TensorRT engine building process also works fine for YOLO-World.