Skip to content

Minor Issue: The notebook tutorial is using the wrong model (yolov11s instead of RF-DETR) #431

Description

@erfan475

Search before asking

  • I have searched the Roboflow Notebooks issues and found no similar bug report.

Notebook name

https://github.com/roboflow/notebooks/blob/main/notebooks/basketball-ai-how-to-detect-track-and-identify-basketball-players.ipynb

Bug

I wanted to play around with RF-DETR in this tutorial and trying your inference library to get the model from roboflow universe then I realized
"basketball-player-detection-3-ycjdo/4" corresponds to a yolov11s and YOLOv11ObjectDetection class
It should change to
"basketball-player-detection-3-ycjdo/13" which corresponds to a rfdetr-medium

Since the links to downloadable models can change and get outdated I suggest adding your prefered method to verify the model is actually RF-DETR architecture.
I currently do print(f"Architecture Handler: {model.__class__.__name__}") to verify my architecture, but it doesn't say which version of the model is being used (S/M/X/XL).
Is there a way to do this in code without having to peek the inference cache meta-data (model_type.json) or browsing the webpage to see which number corresponds to which model?

Environment

  • It's irrelavant here but here you go:
  • Local
  • Python 3.11.9
  • Windows11

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

  • Yes I'd like to help by submitting a PR!

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions