search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.54k stars 496 forks source link

yolo-nas sem #1775

Closed POVTUASTHOV closed 8 months ago

POVTUASTHOV commented 8 months ago

💡 Your Question

i want to learn about yolo nas with segmentation. but I tested with 1 image and got an error, can you help me?

from super_gradients.common.object_names import Models from super_gradients.training import models

model = models.get(model_name=Models.PP_LITE_T_SEG75, pretrained_weights="cityscapes")

IMAGES = [ "https://deci-pretrained-models.s3.amazonaws.com/sample_images/beatles-abbeyroad.jpg", ]

predictions = model.predict(IMAGES) predictions.show() predictions.save(output_folder="") # Save in working directory

AttributeError: 'PPLiteSegT' object has no attribute 'predict'

Versions

No response

Louis-Dupont commented 8 months ago

Hi @POVTUASTHOV, the .predict API for segmentation models is still in our roadmap. Until we release it, you can write a few lines of code to do the same, as shown in this collab notebook (check the segmentation part). Hoping this helps!