From f1a57a36f6019193f515e5e2cfbbabc21f8d4dce Mon Sep 17 00:00:00 2001 From: jamjamjon Date: Thu, 11 Jul 2024 20:44:24 +0800 Subject: [PATCH] update crate doc --- src/lib.rs | 41 +++++++++++++++++++++++------------------ 1 file changed, 23 insertions(+), 18 deletions(-) diff --git a/src/lib.rs b/src/lib.rs index e63e9af..3b8ea2c 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -7,24 +7,29 @@ //! //! # Supported models -//! | Model | Task / Type | Example | CUDA
f32 | CUDA
f16 | TensorRT
f32 | TensorRT
f16 | -//! | :---------------------------------------------------------------: | :-------------------------: | :----------------------: | :-----------: | :-----------: | :------------------------: | :-----------------------: | -//! | [YOLOv5](https://github.com/ultralytics/yolov5) | Object Detection
Instance Segmentation
Classification | [demo](examples/yolov5) | ✅ | ✅ | ✅ | ✅ | -//! | [YOLOv8-obb](https://github.com/ultralytics/ultralytics) | Object Detection
Instance Segmentation
Classification
Oriented Object Detection
Keypoint Detection | [demo](examples/yolov8) | ✅ | ✅ | ✅ | ✅ | -//! | [YOLOv9](https://github.com/WongKinYiu/yolov9) | Object Detection | [demo](examples/yolov9) | ✅ | ✅ | ✅ | ✅ | -//! | [YOLOv10](https://github.com/THU-MIG/yolov10) | Object Detection | [demo](examples/yolov10) | ✅ | ✅ | ✅ | ✅ | -//! | [RT-DETR](https://arxiv.org/abs/2304.08069) | Object Detection | [demo](examples/rtdetr) | ✅ | ✅ | ✅ | ✅ | -//! | [FastSAM](https://github.com/CASIA-IVA-Lab/FastSAM) | Instance Segmentation | [demo](examples/fastsam) | ✅ | ✅ | ✅ | ✅ | -//! | [YOLO-World](https://github.com/AILab-CVC/YOLO-World) | Object Detection | [demo](examples/yolo-world) | ✅ | ✅ | ✅ | ✅ | -//! | [DINOv2](https://github.com/facebookresearch/dinov2) | Vision-Self-Supervised | [demo](examples/dinov2) | ✅ | ✅ | ✅ | ✅ | -//! | [CLIP](https://github.com/openai/CLIP) | Vision-Language | [demo](examples/clip) | ✅ | ✅ | ✅ visual
❌ textual | ✅ visual
❌ textual | -//! | [BLIP](https://github.com/salesforce/BLIP) | Vision-Language | [demo](examples/blip) | ✅ | ✅ | ✅ visual
❌ textual | ✅ visual
❌ textual | -//! | [DB](https://arxiv.org/abs/1911.08947) | Text Detection | [demo](examples/db) | ✅ | ✅ | ✅ | ✅ | -//! | [SVTR](https://arxiv.org/abs/2205.00159) | Text Recognition | [demo](examples/svtr) | ✅ | ✅ | ✅ | ✅ | -//! | [RTMO](https://github.com/open-mmlab/mmpose/tree/main/projects/rtmo) | Keypoint Detection | [demo](examples/rtmo) | ✅ | ✅ | ❌ | ❌ | -//! | [YOLOPv2](https://arxiv.org/abs/2208.11434) | Panoptic Driving Perception | [demo](examples/yolop) | ✅ | ✅ | ✅ | ✅ | -//! | [Depth-Anything](https://github.com/LiheYoung/Depth-Anything) | Monocular Depth Estimation | [demo](examples/depth-anything) | ✅ | ✅ | ❌ | ❌ | -//! | [MODNet](https://github.com/ZHKKKe/MODNet) | Image Matting | [demo](examples/modnet) | ✅ | ✅ | ✅ | ✅ | +//! | Model | Task / Type | +//! | :---------------------------------------------------------------: | :-------------------------: | +//! | [YOLOv5](https://github.com/ultralytics/yolov5) | Object Detection
Instance Segmentation
Classification | +//! | [YOLOv6](https://github.com/meituan/YOLOv6) | Object Detection | +//! | [YOLOv7](https://github.com/WongKinYiu/yolov7) | Object Detection | +//! | [YOLOv8](https://github.com/ultralytics/ultralytics) | Object Detection
Instance Segmentation
Classification
Oriented Object Detection
Keypoint Detection | +//! | [YOLOv9](https://github.com/WongKinYiu/yolov9) | Object Detection | +//! | [YOLOv10](https://github.com/THU-MIG/yolov10) | Object Detection | +//! | [RT-DETR](https://arxiv.org/abs/2304.08069) | Object Detection | +//! | [FastSAM](https://github.com/CASIA-IVA-Lab/FastSAM) | Instance Segmentation | +//! | [YOLO-World](https://github.com/AILab-CVC/YOLO-World) | Object Detection | +//! | [DINOv2](https://github.com/facebookresearch/dinov2) | Vision-Self-Supervised | +//! | [CLIP](https://github.com/openai/CLIP) | Vision-Language | +//! | [BLIP](https://github.com/salesforce/BLIP) | Vision-Language | +//! | [DB](https://arxiv.org/abs/1911.08947) | Text Detection | +//! | [SVTR](https://arxiv.org/abs/2205.00159) | Text Recognition | +//! | [RTMO](https://github.com/open-mmlab/mmpose/tree/main/projects/rtmo) | Keypoint Detection | +//! | [YOLOPv2](https://arxiv.org/abs/2208.11434) | Panoptic Driving Perception | +//! | [Depth-Anything
(v1, v2)](https://github.com/LiheYoung/Depth-Anything) | Monocular Depth Estimation | +//! | [MODNet](https://github.com/ZHKKKe/MODNet) | Image Matting | + +//! # Examples +//! [All Examples Here](https://github.com/jamjamjon/usls/tree/main/examples) //! # Use provided models for inference