GantMan/nsfw_model

floating point number issues

ThisLimn0 opened this issue · 0 comments

E:\GitHub\test.jpg size: (224, 224)
{
"E:\GitHub\test.jpg": {
"drawings": 5.642194810207002e-05,
"hentai": 3.605247547966428e-05,
"neutral": 0.9992769360542297,
"porn": 1.0149957461180747e-06,
"sexy": 0.0006295803468674421
}
}

I'm getting floating point number issues. Maybe add rounding to circumvent this?