Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Resources  
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Explore Pricing
Product  
Data Engine LLMs Platform Enterprise
Connect to Dagshub Discord channel
Register
Login

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git

Watch 4
Issues 62 Pull Requests 44 Discussions 0

#842 fixed version

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:bugfix/SG-000_fix_version
Conversation 0 Files changed 2 Commits 1
File Compare
super-gradients
/
src
/
super_gradients
/
examples
/
predict
/
detection_predict.py
View left file View right file
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
  1. from super_gradients.common.object_names import Models
    2. 

  2. from super_gradients.training import models
    3. 

  3. 

  4. # Note that currently only YoloX and PPYoloE are supported.
    5. 

  5. model = models.get(Models.PP_YOLOE_S, pretrained_weights="coco")
    6. 

  6. 

  7. IMAGES = [
    8. 

  8.     "../../../../documentation/source/images/examples/countryside.jpg",
    9. 

  9.     "../../../../documentation/source/images/examples/street_busy.jpg",
    10. 

  10.     "https://cdn-attachments.timesofmalta.com/cc1eceadde40d2940bc5dd20692901371622153217-1301777007-4d978a6f-620x348.jpg",
    11. 

  11. ]
    12. 

  12. 

  13. prediction = model.predict(IMAGES)
    14. 

  14. prediction.show()
    15.
Discard
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file