changes to deepcolor downloader to tolerate corrupt files. When my ha…

…rd drive ran out of space the script started saving crap
bfortuner · Apr 7, 2017 · 3b65fce · 3b65fce
1 parent 767c908
commit 3b65fce
Show file tree

Hide file tree

Showing 2 changed files with 1,283 additions and 338 deletions.
diff --git a/deepcolor.ipynb b/deepcolor.ipynb
diff --git a/pytorch_cheatsheet.ipynb b/pytorch_cheatsheet.ipynb
@@ -1232,6 +1232,39 @@
     "print(t.min())"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Loading Preexisting Datasets"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "collapsed": true
+   },
+   "outputs": [],
+   "source": [
+    "transform = transforms.Compose(\n",
+    "    [transforms.ToTensor(),\n",
+    "     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])\n",
+    "\n",
+    "trainset = torchvision.datasets.CIFAR10(root='./data', train=True,\n",
+    "                                        download=True, transform=transform)\n",
+    "trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,\n",
+    "                                          shuffle=True, num_workers=2)\n",
+    "\n",
+    "testset = torchvision.datasets.CIFAR10(root='./data', train=False,\n",
+    "                                       download=True, transform=transform)\n",
+    "testloader = torch.utils.data.DataLoader(testset, batch_size=4,\n",
+    "                                         shuffle=False, num_workers=2)\n",
+    "\n",
+    "classes = ('plane', 'car', 'bird', 'cat',\n",
+    "           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -1467,6 +1500,12 @@
     "    \"\"\"Imshow for Tensor.\"\"\"\n",
     "    inp = decode_image(tensor)\n",
     "    plt.imshow(inp)\n",
+    "    plt.show()\n",
+    "    \n",
+    "def plot_img_array(img_arr, shape=(16,16)):\n",
+    "    # Plot numpy array image with shape (inches?)\n",
+    "    plt.figure(figsize=shape)\n",
+    "    plt.imshow(img_arr.astype('uint8'))\n",
     "    plt.show()"
    ]
   },
@@ -1776,6 +1815,26 @@
    "source": [
     "view_sample_predictions(3)"
    ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## References\n",
+    "\n",
+    "* https://github.com/desimone/pytorch-cat-vs-dogs/blob/master/main.py"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "collapsed": true
+   },
+   "outputs": [],
+   "source": [
+    "t1 = torch."
+   ]
   }
  ],
  "metadata": {