Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

integrate yolo as ML backend into my label tasks #6825

Open
JasonSloan opened this issue Dec 25, 2024 · 1 comment
Open

integrate yolo as ML backend into my label tasks #6825

JasonSloan opened this issue Dec 25, 2024 · 1 comment

Comments

@JasonSloan
Copy link

Hi, i tried to integrate the ML models into my label tasks, everything seemd to work fine but no predictions showed though.

Env: label-studio was running in a docker container while the ML-backend was running in the container of the container, but i checked the ENV variable as well as the mapping directory inside the ML-backend container and they were all correctly set.

Tests: I tested in my web browser of the connection between label-studio and ML-backend and it worked fine.
image

Model weight: I put an official yolov8m.pt file downloaded from github into the models folder and i checked inside the ML-backend container that it was really being there.
image

Settings: I set the label template to yolo and wrote the configurations code below:
<View> <Image name="image" value="$image"/> <RectangleLabels name="label" toName="image" model_score_threshold="0.25" model_path="[yolov8n.pt](http://yolov8n.pt/)" opacity="0.1"> <Label value="Person" background="red"/> <Label value="person" background="red"/> <Label value="Car" background="blue"/> </RectangleLabels> </View>

Finally, i checked the images inside the project but no person bounding boxes were shown. Besides, I was pretty sure that the yolov8m.pt can successfully detected them.

So would you please tell me that where i was doing wrong?

@makseq
Copy link
Member

makseq commented Dec 29, 2024

Hi,

  1. This line looks wrong in your labeling config
 model_path="[yolov8n.pt](http://yolov8n.pt/)"

it should be

model_path="yolov8n.pt"
  1. You don't need to download official models manually, they will be downloaded automatically.
  2. Most likely, you didn't specify LABEL_STUDIO_URL and LABEL_STUDIO_API_KEY in the docker-compose.yml file.
  3. Check ML backend logs, the problem should be mentioned there.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants