roboflow / roboflow/notebooks

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

Open
#431 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

bug
Dominant language
Jupyter Notebook
Stars
9.7k
Forks
1.5k
PR merge metrics
No merged PRs in 30d

Description

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!

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Open notebooks/basketball-ai-how-to-detect-track-and-identify-basketball-players.ipynb and locate the model identifier and YOLOv11ObjectDetection references. Update the tutorial to use basketball-player-detection-3-ycjdo/13 and the corresponding RF-DETR model, then run the affected notebook cells to confirm the tutorial loads the intended architecture. The architecture-version verification question remains additional scope.

Written by the indexing model from the issue text.

Assessment

Tech stack
jupyter-notebook, python
Domain
computer-vision, documentation
Issue type
Bug
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
52/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.