Skip to content
This repository has been archived by the owner on Dec 14, 2023. It is now read-only.

create a process for tensorflow model training and classification #707

Open
cindyloo opened this issue May 19, 2020 · 0 comments
Open

create a process for tensorflow model training and classification #707

cindyloo opened this issue May 19, 2020 · 0 comments

Comments

@cindyloo
Copy link

we can use tensorflow, keras layers and a pre-trained model (like ResNet or MobileNet) to classify story images. We can adapt pre-existing models or create our own - it depends on the type of images we care to analyze. As ImageNet is trained on millions of images, a good first analysis would be to instantiate the model, grab feature extractions, and train a top level layer on our own dataset. This dataset would be a subset of topic images that we would then use to make predictions on the rest of the images within that topic

Generally the pseudo code would look something like this:

from tensorflow.keras import layers
from tensorflow import keras
import tensorflow as tf

#gather a dataset from either our own downloaded topic images or a tensorflow dataset
# reshape the images to a numPy array as tensors w/ standard size, likely 28 x 28
# load a pre-trained ResNet model or our own "keras.h5" 
#re-train the model on our dataset
#use that model to evaluate another dataset according to our interests
#save the updated model for next use
#optionally store those predictions
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant