ds img

Author: slegroux

nbs/image.datasets.ipynb

@@ -1,15 +1,5 @@
 {
  "cells": [
-  {
-   "cell_type": "raw",
-   "metadata": {},
-   "source": [
-    "---\n",
-    "skip_exec: true\n",
-    "skip_showdoc: true\n",
-    "---"
-   ]
-  },
   {
    "attachments": {},
    "cell_type": "markdown",
@@ -79,6 +69,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "#| export\n",
     "set_seed(42)\n",
     "logger = logging.getLogger(__name__)\n",
     "plt.set_loglevel('INFO')"

nimrod/image/datasets.py

@@ -3,9 +3,9 @@
 # AUTOGENERATED! DO NOT EDIT! File to edit: ../../nbs/image.datasets.ipynb.
 
 # %% auto 0
-__all__ = ['ImageDataset', 'MNISTDataset', 'MNISTDataModule']
+__all__ = ['logger', 'ImageDataset', 'MNISTDataset', 'MNISTDataModule']
 
-# %% ../../nbs/image.datasets.ipynb 4
+# %% ../../nbs/image.datasets.ipynb 3
 import torch
 import torch.utils.data as data
 from torch.utils.data import ConcatDataset, DataLoader, Dataset, random_split
@@ -30,7 +30,12 @@
 import logging
 
 
-# %% ../../nbs/image.datasets.ipynb 7
+# %% ../../nbs/image.datasets.ipynb 4
+set_seed(42)
+logger = logging.getLogger(__name__)
+plt.set_loglevel('INFO')
+
+# %% ../../nbs/image.datasets.ipynb 6
 class ImageDataset(Dataset):
     " Base class for image datasets providing visualization of (image, label) samples"
 
@@ -80,7 +85,7 @@ def show_random(
         self.show_grid(images)
 
 
-# %% ../../nbs/image.datasets.ipynb 11
+# %% ../../nbs/image.datasets.ipynb 10
 class MNISTDataset(ImageDataset):
     "MNIST digit dataset"
 
@@ -132,7 +137,7 @@ def train_dev_split(
 
 
 
-# %% ../../nbs/image.datasets.ipynb 18
+# %% ../../nbs/image.datasets.ipynb 17
 class MNISTDataModule(DataModule, LightningDataModule):
     def __init__(self,
                  data_dir: str | os.PathLike = "~/Data/", # path to source data dir