GantMan / GantMan/nsfw_model

New class training and data

Open
#37 8 comments 0 reactions 0 assignees View on GitHub
enhancement help wanted training
Dominant language
Python
Stars
2.1k
Forks
303
PR merge metrics
No merged PRs in 30d

Description

Hello!

Im a ML practitioner from Chile (so im sorry in advance for any english mistakes). I was thinking about improving this model with a new NSFW class: violence (gore for example). I understand basic concepts of transfer learning and how to do it, but since you are the expert I would like to know if you can help me with:

1.- How much data of the original classes would you use for this new training? Should I use the 60 gbs or that would be overkill since the model already can already classify the 5 original classes.
2.- How much data do you think i need for the new class?

Btw im so grateful for your code, especially the mobilenet version! how long did it take to train that model?

Cya!

Contributor guide

No contributing guide indexed for this repository

Research direction

No files, tests, or entry points are named. Start by reviewing the repository's Keras model and training-data setup, then determine how the five existing classes and a proposed violence class are represented. Done would require an agreed training and dataset plan rather than a specific code change.

Written by the indexing model from the issue text.

Assessment

Tech stack
keras, python
Domain
data, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.