Skip to content

Instantly share code, notes, and snippets.

View Marina-Miranovich's full-sized avatar

Marina Miranovich Marina-Miranovich

View GitHub Profile
@Marina-Miranovich
Marina-Miranovich / code_snippet_3.js
Created November 14, 2018 20:42
code_snippet_3 for "Playing Mortal Kombat with TensorFlow.js. Transfer learning and data augmentation" translation
export const loadModel = async () => {
const mn = new mobilenet.MobileNet(1, 1);
mn.path = `file://PATH/TO/model.json`;
await mn.load();
return (input): tf.Tensor1D =>
mn.infer(input, 'global_average_pooling2d_1')
.reshape([1024]);
};
@Marina-Miranovich
Marina-Miranovich / code_snippet_2.js
Created November 14, 2018 20:37
code_snippet_2 for "Playing Mortal Kombat with TensorFlow.js. Transfer learning and data augmentation" translation
import * as tf from '@tensorflow/tfjs';
const model = tf.sequential();
model.add(tf.layers.inputLayer({ inputShape: [1024] }));
model.add(tf.layers.dense({ units: 1024, activation: 'relu' }));
model.add(tf.layers.dense({ units: 1, activation: 'sigmoid' }));
model.compile({
optimizer: tf.train.adam(1e-6),
loss: tf.losses.sigmoidCrossEntropy,
metrics: ['accuracy']
@Marina-Miranovich
Marina-Miranovich / 1.py
Created November 14, 2018 20:02
code_snippet_1 for "Playing Mortal Kombat with TensorFlow.js. Transfer learning and data augmentation" translation
np.random.seed(44)
ia.seed(44)
def main():
for i in range(1, 191):
draw_single_sequential_images(str(i), "others", "others-aug")
for i in range(1, 191):
draw_single_sequential_images(str(i), "hits", "hits-aug")
for i in range(1, 191):
draw_single_sequential_images(str(i), "kicks", "kicks-aug")
def draw_single_sequential_images(filename, path, aug_path):