View file src/colab/deci_yolo_nas.py - Download

# -*- coding: utf-8 -*-
"""Deci-YOLO-NAS.ipynb

Automatically generated by Colaboratory.

Original file is located at
    https://colab.research.google.com/drive/1RgUcXlX-j5O8PKgB1Au7A4KKAZ0yy2RF

https://deci.ai/model-zoo/yolo-nas/
"""

!pip install super_gradients
import super_gradients
yolo_nas = super_gradients.training.models.get("yolo_nas_l", pretrained_weights="coco").cuda()
yolo_nas.predict("https://deci-pretrained-models.s3.amazonaws.com/sample_images/beatles-abbeyroad.jpg").show()