Instructions to use riversnow/so101-segmentation-model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- ultralytics
How to use riversnow/so101-segmentation-model with ultralytics:
# Couldn't find a valid YOLO version tag. # Replace XX with the correct version. from ultralytics import YOLOvXX model = YOLOvXX.from_pretrained("riversnow/so101-segmentation-model") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True) - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -24,7 +24,7 @@ This is a model for segementation of images of the [so101 robot arm](https://git
|
|
| 24 |
|
| 25 |
Here's some sample code to use it
|
| 26 |
|
| 27 |
-
```
|
| 28 |
import cv2
|
| 29 |
import numpy as np
|
| 30 |
from ultralytics import YOLO
|
|
|
|
| 24 |
|
| 25 |
Here's some sample code to use it
|
| 26 |
|
| 27 |
+
```python
|
| 28 |
import cv2
|
| 29 |
import numpy as np
|
| 30 |
from ultralytics import YOLO
|