Merge pull request #15 from nekitmm/tf2

Tf2

README.md

@@ -7,6 +7,9 @@ I also created a few Jupyter notebooks for ease of use:
 1. starnet_v1_TF2_transform.ipynb - loads and transforms an image.
 2. starnet_v1_TF2.ipynb - more detailed example that loads a model and shows how to train it (really simple as well I think).
 
+Weights for the new model can be found <a href="https://www.dropbox.com/s/lcgn5gvnxpo27s5/starnet_weights2.zip?dl=0">here</a>.
+
+<hr>
 
 **StarNet** is a neural network that can remove stars from images in one simple step leaving only background.
 

starnet_v1_TF2.ipynb

@@ -160,6 +160,18 @@
    "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,

starnet_v1_TF2_transform.ipynb

@@ -13,6 +13,13 @@
     "from starnet_v1_TF2 import StarNet"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<p>Load a model. There are two modes - RGB and Greyscale.</p>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 12,
@@ -23,6 +30,13 @@
     "starnet.load_model('./weights', './history')"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<p>Load and transform an image.</p>"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 13,