sugarBecomer
11 lines
313 B
Go
11 lines
313 B
Go
package classifier
|
|
|
|
var imageClassifier, err = NewImageClassifier("./static/model.onnx", "./static/classes.txt")
|
|
|
|
// 图片数据 & 类别名称
|
|
// return 是否匹配
|
|
func ClassifyImage(imgData []byte, className string) bool {
|
|
name, _, _ := imageClassifier.PredictFromBytes(imgData)
|
|
return className == name
|
|
}
|