Released in 2016 by Yahoo, this net is a binary classifier (safe/not safe) fine-tuned from the pre-trained ResNet-50 model. It determines whether an image is not suitable/safe for work (NSFW), due to the presence of nudity and/or pornographic content. The original model was chosen to provide a good tradeoff between accuracy and computational weight.
Number of layers: 177 |
Parameter count: 5,944,514 |
Trained size: 24 MB |
Examples
Resource retrieval
Get the pre-trained net:
Out[1]= |  |
Basic usage
Apply the trained net to an input image:
Out[2]= |  |
Obtain the probabilities for each class for two images:
Out[3]= |  |
Net information
Inspect the number of parameters of all arrays in the net:
Out[4]= |  |
Obtain the total number of parameters:
Out[5]= |  |
Obtain the layer type counts:
Out[6]= |  |
Display the summary graphic:
Out[7]= |  |
Export to MXNet
Export the net into a format that can be opened in MXNet:
Out[8]= |  |
Export also creates a net.params file containing parameters:
Out[9]= |  |
Get the size of the parameter file:
Out[10]= |  |
The size is similar to the byte count of the resource object:
Out[11]= |  |
Requirements
Wolfram Language 11.2
(September 2017) or above
External Links
Resource History
Reference