iOS Core ML keras added

iOSCoreMLKeras/ml/.ipynb_checkpoints/Untitled1-checkpoint.ipynb

@@ -0,0 +1,70 @@
+{
+ "cells": [
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "Found 2001 images belonging to 2 classes.\n",
+ "{'cats': 0, 'dogs': 1}\n",
+ "Found 1008 images belonging to 2 classes.\n",
+ "Epoch 1/50\n"
+ ]
+ }
+ ],
+ "source": [
+ "import train"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 5,
+ "metadata": {},
+ "outputs": [
+ {
+ "name": "stdout",
+ "output_type": "stream",
+ "text": [
+ "['dogs', 'cats']\n"
+ ]
+ }
+ ],
+ "source": [
+ "import os\n",
+ "print(os.listdir('data/train'))"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": []
+ }
+ ],
+ "metadata": {
+ "kernelspec": {
+ "display_name": "Python 3",
+ "language": "python",
+ "name": "python3"
+ },
+ "language_info": {
+ "codemirror_mode": {
+ "name": "ipython",
+ "version": 3
+ },
+ "file_extension": ".py",
+ "mimetype": "text/x-python",
+ "name": "python",
+ "nbconvert_exporter": "python",
+ "pygments_lexer": "ipython3",
+ "version": "3.7.3"
+ }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

iOSCoreMLKeras/ml/coremltoolconverter.py

@@ -0,0 +1,12 @@
+import coremltools
+
+coreml_model = coremltools.converters.keras.convert('model.h5', input_names=['image'], output_names=['output'],
+ image_input_names='image')
+
+coreml_model.author = 'Anupam Chugh'
+coreml_model.short_description = 'Cat Dog Classifier converted from a Keras model'
+coreml_model.input_description['image'] = 'Takes as input an image'
+coreml_model.output_description['output'] = 'Prediction as cat or dog'
+
+
+coreml_model.save('catdogcoreml.mlmodel')