DeepStream-Yolo icon indicating copy to clipboard operation
DeepStream-Yolo copied to clipboard

Add support for YOLO-World

Open YuhyunNa opened this issue 11 months ago • 3 comments

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

YuhyunNa avatar Mar 13 '24 02:03 YuhyunNa

@marcoslucianops

YuhyunNa avatar Mar 25 '24 04:03 YuhyunNa

Have you completed the TensorRT acceleration for YOLO-World? If so, that's really great work.

lzylzylzy123456 avatar Mar 28 '24 07:03 lzylzylzy123456

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.

YuhyunNa avatar Apr 08 '24 02:04 YuhyunNa