Python NSFW 检测模块 nudenet 不再工作?

Python NSFW detection module nudenet not longer working?

我一直在使用 python 模块 nudenet 作为我的最终学位项目。我正在使用 google colab 来 运行 它。

在过去的几个月里,它工作正常,没有任何问题,直到昨天,当我尝试导入它时,这个错误发生了:

!pip install --upgrade nudenet
from nudenet import NudeClassifier

ImportError: cannot import name '_registerMatType' from 'cv2.cv2' (/usr/local/lib/python3.7/dist-packages/cv2/cv2.cpython-37m-x86_64-linux-gnu.so)

我试图通过将 opencv-python-headless 降级到以前的版本

来解决这个错误
!pip uninstall opencv-python-headless==4.5.5.62 
!pip install opencv-python-headless==4.5.1.48

但是,当我加载分类器时出现此错误:

classifier = NudeClassifier()


Downloading the checkpoint to /root/.NudeNet/classifier_model.onnx
MB| |#                                                | 0 Elapsed Time: 0:00:00
Content-length not found, file size cannot be estimated.
Succefully Downloaded to: /root/.NudeNet/classifier_model.onnx

InvalidProtobuf: [ONNXRuntimeError] : 7 : INVALID_PROTOBUF : Load model from /root/.NudeNet/classifier_model.onnx failed:Protobuf parsing failed.

我也试过降级nudenet模块的版本,还是不行。

提前致谢。

This is the link for the module in github

将规则 classifier_model.onnx 替换为 nudenet classifier 并用 nudenet lite classifier

归档 classifier_lite.onnx