Python NSFW 检测模块 Nudenet 不再工作？

Question

我一直在我的最终学位项目中使用 python 模块nudnet 。我正在使用 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模块的版本，但仍然不起作用。

先感谢您。

这是 github 中模块的链接

Answer 1

将 rile classifier_model.onnx 替换为Nudenet 分类器 ，将文件 classifier_lite.onnx 替换为Nudenet lite 分类器