Skip to content

Commit

Permalink
docs(README): [0.1.0] Update download link
Browse files Browse the repository at this point in the history
  • Loading branch information
Steeeephen authored Oct 10, 2022
1 parent a870125 commit f7f8b0a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,11 @@

This is the repo for Bumble's *Private Detector*™ model - an image classifier that can detect lewd images.

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
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)

## Model

The SavedModel can be found in `saved_model/`, ensure you have [Git LFS](https://git-lfs.github.com/) installed to pull the full files.
The SavedModel can be found in `saved_model/` within `private_detector.zip` above

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)

Expand Down Expand Up @@ -49,7 +49,7 @@ python3 inference.py \

## Additional Training

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/`
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`

Set up a JSON file with links to your image path lists for each class:

Expand Down

0 comments on commit f7f8b0a

Please sign in to comment.