Draw the sketch for ADDA.

franck-armand · Aug 17, 2017 · ebb5a76 · ebb5a76
1 parent 538965f
commit ebb5a76
Show file tree

Hide file tree

Showing 11 changed files with 44 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,10 @@
+# pytorch dataset
+data
+model
+
+# trash
+.DS_Store
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

diff --git a/README.md b/README.md
@@ -1,2 +1,2 @@
-# pytorch-adda
-PyTorch implementation for Adversarial Discriminative Domain Adaptation
+# PyTorch-ADDA
+A PyTorch implementation for [Adversarial Discriminative Domain Adaptation](https://arxiv.org/abs/1702.05464).
diff --git a/core/__init__.py b/core/__init__.py
diff --git a/core/adapt.py b/core/adapt.py
@@ -0,0 +1 @@
+"""Adversarial adaptation to train target encoder."""
diff --git a/core/pretrain.py b/core/pretrain.py
@@ -0,0 +1 @@
+"""Pre-train encoder and classifier for source dataset."""
diff --git a/core/test.py b/core/test.py
@@ -0,0 +1 @@
+"""Test script to classify target data."""
diff --git a/datasets/__init__.py b/datasets/__init__.py
diff --git a/datasets/mnist.py b/datasets/mnist.py
@@ -0,0 +1,29 @@
+"""Dataset setting and data loader for MNIST."""
+
+
+import torch
+from torchvision import datasets, transforms
+
+import params
+
+# image pre-processing
+pre_process = transforms.Compose([transforms.ToTensor(),
+                                  transforms.Normalize(
+                                      mean=params.dataset_mean,
+                                      std=params.dataset_std)])
+
+# dataset and data loader
+mnist_dataset = datasets.mnist(root=params.data_root,
+                               transform=pre_process,
+                               download=True)
+
+mnist_data_loader = torch.utils.data.DataLoader(dataset=mnist_dataset,
+                                                batch_size=params.batch_size,
+                                                shuffle=True)
+
+
+def get_mnist():
+    """Inf data iterator."""
+    while True:
+        for images, _ in mnist_data_loader:
+            yield images
diff --git a/datasets/usps.py b/datasets/usps.py
@@ -0,0 +1 @@
+"""Dataset setting and data loader for USPS."""
diff --git a/main.py b/main.py
@@ -0,0 +1 @@
+"""Main script for ADDA."""
diff --git a/utils.py b/utils.py
@@ -0,0 +1 @@
+"""Utilities for ADDA."""
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		"""Adversarial adaptation to train target encoder."""
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		"""Pre-train encoder and classifier for source dataset."""
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		"""Dataset setting and data loader for USPS."""