search

anuragxel / salt

Segment Anything Labelling Tool
MIT License
1.03k stars 125 forks source link

A bug occur when I want to change the categories #53

Open Nimiaoye opened 6 months ago

Nimiaoye commented 6 months ago

I want to change the categories from 'cat' and 'dog' to other labels, but as I want to change the categories with the command: python segment_anything_annotator.py --onnx-model-path sam_onnx.onnx --dataset-path ../shot --categories i,j,k,l,o

It turns out that the categories in the interface are still cat and dog, and I cannot change the categories whatever I type in. So I wonder if someone who once faced the same problem can give me an explanation and help me slove this problem, thank you so much!!! ^-^

Skwarson96 commented 4 months ago

From what I remember, the categories are taken from the annotations.json file. This file is created immediately after running the script and is only loaded later. So either you can change the category names in this file or delete it and it will be re-created with new category names (but you will lose the annotations you drew earlier)