Peter Viergutz peterviergutz

Retrain a MobileNet model and use it in the browser with TensorFlow.js

DRAFT

Combining TensorFlow for Poets and TensorFlow.js.
Retrain a MobileNet V1 or V2 model on your own dataset using the CPU only.
I'm using a MacBook Pro without Nvidia GPU.

MobileNets can be used for image classification. This guide shows the steps I took to retrain a MobileNet on a custom dataset, and how to convert and use the retrained model in the browser using TensorFlow.js. The total time to set up, retrain the model and use it in the browser can take less than 30 minutes (depending on the size of your dataset).

Example app - HTML/JS and a retrained MobileNet V1/V2 model.

	Geonames Feature Code,Geonames Label,Wikidata URI,Wikidata label,AAT ID,AAT label,description,Geonames Feature Class,Wikipedia Code
	ADM1,first-order administrative division,http://www.wikidata.org/entity/Q10864048,first-level administrative country subdivision,http://vocab.getty.edu/aat/300387064,first level subdivisions (political entities),"a primary administrative division of a country, such as a state in the United States",A,A.ADM1
	ADM1H,historical first-order administrative division,,,,,a former first-order administrative division,A,A.ADM1H
	ADM2,second-order administrative division,http://www.wikidata.org/entity/Q13220204,second-level administrative country subdivision,http://vocab.getty.edu/aat/300387145,second level subdivisions (political entities),a subdivision of a first-order administrative division,A,A.ADM2
	ADM2H,historical second-order administrative division,,,,,a former second-order administrative division,A,A.ADM2H
	ADM3,third-order administrative division,http://www.wikidata.org/entity/Q132217

	/**
	* Super simple encryption utilities
	* Inspired by https://gist.github.com/sukima/5613286
	* Properly handles UTF-8 strings
	* Use these functions at your own risk: xor encryption provides only acceptable
	* security when the key is random and longer than the message
	* If looking for more reliable security use: https://tweetnacl.js.org/
	*
	* Use passwordDerivedKey function in this file to generate a key from a password, or to generate a random key
	*/

	-- Create the raw events table
	CREATE TABLE page_views (
	site_id int,
	path text,
	client_ip inet,
	view_time timestamptz default now(),
	view_id bigserial
	);

	-- Allow fast lookups of ranges of sequence IDs

	import babel from 'rollup-plugin-babel';
	import eslint from 'rollup-plugin-eslint';
	import resolve from 'rollup-plugin-node-resolve';
	import multiEntry from 'rollup-plugin-multi-entry';
	import uglify from 'rollup-plugin-uglify';
	import filesize from 'rollup-plugin-filesize';
	import commonjs from 'rollup-plugin-commonjs';
	import progress from 'rollup-plugin-progress';

	let pluginOptions = [

	from __future__ import print_function
	import imageio
	from PIL import Image
	import numpy as np
	import keras

	from keras.layers import Input, Dense, Conv2D, MaxPooling2D, AveragePooling2D, ZeroPadding2D, Dropout, Flatten, Concatenate, Reshape, Activation
	from keras.models import Model
	from keras.regularizers import l2
	from keras.optimizers import SGD

	// XORCipher - Super simple encryption using XOR and Base64
	//
	// Depends on [Underscore](http://underscorejs.org/).
	//
	// As a warning, this is not a secure encryption algorythm. It uses a very
	// simplistic keystore and will be easy to crack.
	//
	// The Base64 algorythm is a modification of the one used in phpjs.org
	// * http://phpjs.org/functions/base64_encode/
	// * http://phpjs.org/functions/base64_decode/

	/*

	wget http://dl.dropbox.com/u/342088/10m-us-parks-area.sqlite

	ogr2ogr -f SQLite parks.sqlite ParkPly_900913.shp

	dane$ ogr2ogr -f SQLite parks.sqlite ~/Documents/MapBox/cache/2976a0c1-10m-us-parks-area/2976a0c1-10m-us-parks-area.shp

	*/

	root=/path/to/root
	project=/path/to/project