search

vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything, MobileSAM!!
https://anylabeling.nrl.ai
GNU General Public License v3.0
2k stars 218 forks source link

the problem about load SAM autolabeling #158

Closed fatejzz closed 1 week ago

fatejzz commented 7 months ago

while I load the offical model like SAM(download from https://github.com/vietanhdev/anylabeling-assets/releases/tag/v0.4.0) in custom model way,the program will shut down,and the console says"[ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: x Got: 3 Expected: 4 Please fix either the inputs or the model."

fscbird commented 7 months ago

I also has this probles...

vietanhdev commented 6 months ago

Please use the samexporter to export ONNX models for the latest version of AnyLabeling. Thank you!