Test code for data without label

BAAI-DCAI / SegVol

The official code for "SegVol: Universal and Interactive Volumetric Medical Image Segmentation".

MIT License

245 stars 22 forks source link

Test code for data without label #11

Closed jianjun0407 closed 10 months ago

jianjun0407 commented 10 months ago

Tried the web version of the test you provided（https://huggingface.co/spaces/BAAI/SegVol）, can support test data without label. But the inferce_demo.py test code on github can only test data with labels. Could you expose the test code of the data without label and test it by the user entering the prompt point coordinates? In other words, can the test code called by the web version be exposed? Looking forward to your reply!

Yuxin-Du-Lab commented 10 months ago

Actually the web demo code is also open-source. You can see the code in the SegVol space/Files tag and clone it from here (note that the web demo code is run on CPU):

20240109094103