matrix-org / matrix-org/mjolnir

Consider using a more accurate nsfwjs model

Open
#562 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
TypeScript
Stars
445
Forks
75
Avg merge
2d 44m
Merged PRs (30d)
3

Description

The default one seems slightly inaccurate in some cases, for example the default model identifies this image of my element web call stack as a high chance of being porn
| Default | MobileNetV2Mid |
|--------|--------|
| ![image](https://github.com/user-attachments/assets/3430b624-5bb5-4b69-b56e-c1aca5ba3e19) | ![image](https://github.com/user-attachments/assets/03be3420-5f1d-4292-9571-9e2b91ec80d7) |

In particular, i suggest changing to https://github.com/infinitered/nsfwjs?tab=readme-ov-file#load-the-model "MobileNetV2Mid", the InceptionV3 model might be interesting too, but not sure how it would affect size

Contributor guide

Open the contributing guide

Research direction

Start by locating where mjolnir loads or configures the nsfwjs model, then read nsfwjs's model-loading guidance for MobileNetV2Mid and InceptionV3. Compare the current default with the proposed model for classification accuracy and size; done means a justified default choice is implemented and verified against the reported false positive.

Written by the indexing model from the issue text.

Assessment

Tech stack
typescript
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.