Skip to content

Commit

Permalink
Remove TF1 vs TF2 comparison section, which is no longer relevant
Browse files Browse the repository at this point in the history
  • Loading branch information
bhky committed Jul 19, 2024
1 parent 9a4f78e commit d2a9cdf
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 23 deletions.
23 changes: 0 additions & 23 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -252,26 +252,3 @@ This implementation provides the following preprocessing options.
- Convert to a NumPy image.
- Swap the colour channels to BGR.
- Subtract the training dataset mean value of each channel: `[104, 117, 123]`.

## Comparison

Using 521 private test images, the NSFW probabilities given by
three different settings are compared:
- [TensorFlow 1 implementation](https://github.com/mdietrichstein/tensorflow-open_nsfw) with `YAHOO` preprocessing.
- Keras implementation with `YAHOO` preprocessing.
- Keras implementation with `SIMPLE` preprocessing.

The following figure shows the result:

![NSFW probabilities comparison](docs/nsfw_probabilities_comparison.png)

The current Keras implementation with `YAHOO` preprocessing
can totally reproduce the well-tested TensorFlow 1 result,
with small floating point errors only.

With `SIMPLE` preprocessing the result is different, where the model tends
to give lower probabilities over the current test images.

Note that this comparison does not conclude which preprocessing method is
"better", it only shows their discrepancies. However, users that prefer the
original Yahoo result should go for the default `YAHOO` preprocessing.
Binary file removed docs/nsfw_probabilities_comparison.png
Binary file not shown.

0 comments on commit d2a9cdf

Please sign in to comment.