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

Question

我一直在使用 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

Answer 1

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

归档 classifier_lite.onnx

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

Python NSFW detection module nudenet not longer working?

python

google-colaboratory

onnx