search

CaptchaAgent / hcaptcha-model-factory

🏗 hCaptcha image label binary model factory (PyTorch Training, Cluster-based Auto Label Tools, Export ONNX model, ONNX model inference)
GNU General Public License v3.0
72 stars 18 forks source link

Predict image with YOLO model " yolov5n6.onnx '' #10

Closed hoangthangbk51 closed 2 years ago

hoangthangbk51 commented 2 years ago

Dear sir . Can you have me to classification the image with model "yolov5n6.onnx" I try the other is return the result "0" or "1" But with this model i don't know witch way to do . if possible please give me example code . thanks for support .

beiyuouo commented 2 years ago

No, yolo is for object detection task. It will return the labels, instead of 0 or 1. You can run yolo model follow this This repo is only for classification task, which not contains yolo model, sorry.