wkpark / uddetailer

μ DDetailer, DDetailer fork to support DDetailer as an extension
GNU Affero General Public License v3.0
57 stars 2 forks source link

cannot load onnx model file (nsfwrecog_v1) #85

Open sjkoo1989 opened 4 months ago

sjkoo1989 commented 4 months ago

It looks like my uddetailer plugin cannot load an onnx model file.

list

image presented below is my current setting for webui

version

wkpark commented 4 months ago

image

image

nsfwrecog_v1 will not downloaded automatically, you have to download it with Download helper menu in the uddetailer

sjkoo1989 commented 4 months ago

r

I have already downloaded the model with model download manager. I also copied the model into mmdet/bbox folder. It looks like the system cannot recognize .onnx file, specifically.

capture2 capture3

wkpark commented 4 months ago

nsfwrecog_v1.onnx is a yolov8 model and not supported by mmdet. you should save/move nsfwrecog_v1.onnx into models/yolo folder and remove nsfwrecog_v1.onnx in the mmdet/bbox/ folder