You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardexpand all lines: README.md
+3-3
Original file line number
Diff line number
Diff line change
@@ -2,11 +2,11 @@
2
2
3
3
This is the repo for Bumble's *Private Detector*™ model - an image classifier that can detect lewd images.
4
4
5
-
The internal repo has been heavily refactored and released as a fully open-source project to allow for the wider community to use and finetune a Private Detector model of their own
5
+
The internal repo has been heavily refactored and released as a fully open-source project to allow for the wider community to use and finetune a Private Detector model of their own. You can download the pretrained SavedModel and checkpoint [here](https://storage.googleapis.com/private_detector/private_detector.zip)
6
6
7
7
## Model
8
8
9
-
The SavedModel can be found in `saved_model/`, ensure you have [Git LFS](https://git-lfs.github.com/) installed to pull the full files.
9
+
The SavedModel can be found in `saved_model/` within `private_detector.zip` above
10
10
11
11
The model is based on Efficientnet-v2 and trained on our internal dataset of lewd images - more information can be found at the whitepaper [here](pending)
12
12
@@ -49,7 +49,7 @@ python3 inference.py \
49
49
50
50
## Additional Training
51
51
52
-
You can finetune the model yourself on your own data, to do so is fairly simple - though you will need the checkpoint files as can be found in `saved_checkpoint/`
52
+
You can finetune the model yourself on your own data, to do so is fairly simple - though you will need the checkpoint files as can be found in `saved_checkpoint/` in `private_detector.zip`
53
53
54
54
Set up a JSON file with links to your image path lists for each class:
0 commit comments