GantMan / nsfw_model

Keras model of NSFW detector
Other
1.8k stars 279 forks source link

Script for converting images & GPU #120

Open DankMemeGuy opened 1 year ago

DankMemeGuy commented 1 year ago
  1. Is there any script you used to convert the images? I want to make sure I convert the images with the right settings before I pump it into the training.
  2. I was going through the script, do the scripts use GPU? I noticed there was no direct mention of GPU/CPU so wasn't sure.

Thanks!