NVlabs · sjinu96 · Nov 21, 2021 · Nov 21, 2021 · Nov 27, 2021 · Dec 5, 2021
diff --git a/.gitignore b/.gitignore
diff --git a/.ipynb_checkpoints/gm_Generate New Label-checkpoint.ipynb b/.ipynb_checkpoints/gm_Generate New Label-checkpoint.ipynb
diff --git a/.ipynb_checkpoints/gm_Label Check-checkpoint.ipynb b/.ipynb_checkpoints/gm_Label Check-checkpoint.ipynb
diff --git a/.ipynb_checkpoints/gm_TestTimeTrainingTest-checkpoint.ipynb b/.ipynb_checkpoints/gm_TestTimeTrainingTest-checkpoint.ipynb
diff --git a/.ipynb_checkpoints/gm_Visualize-checkpoint.ipynb b/.ipynb_checkpoints/gm_Visualize-checkpoint.ipynb
diff --git a/data/__init__.py b/data/__init__.py
diff --git a/data/base_dataset.py b/data/base_dataset.py
diff --git a/data/image_folder.py b/data/image_folder.py
diff --git a/data/pix2pix_dataset.py b/data/pix2pix_dataset.py
@@ -29,7 +29,6 @@ def initialize(self, opt):
         label_paths = label_paths[:opt.max_dataset_size]
         image_paths = image_paths[:opt.max_dataset_size]
         instance_paths = instance_paths[:opt.max_dataset_size]
-
         if not opt.no_pairing_check:
             for path1, path2 in zip(label_paths, image_paths):
                 assert self.paths_match(path1, path2), \

diff --git a/datasets/coco_stuff/train_img/000000017914.jpg b/datasets/coco_stuff/train_img/000000017914.jpg
diff --git a/datasets/coco_stuff/train_img/000000029286.jpg b/datasets/coco_stuff/train_img/000000029286.jpg
diff --git a/datasets/coco_stuff/train_img/000000138805.jpg b/datasets/coco_stuff/train_img/000000138805.jpg
diff --git a/datasets/coco_stuff/train_img/000000184101.jpg b/datasets/coco_stuff/train_img/000000184101.jpg
diff --git a/datasets/coco_stuff/train_img/000000197384.jpg b/datasets/coco_stuff/train_img/000000197384.jpg
diff --git a/datasets/coco_stuff/train_img/000000203744.jpg b/datasets/coco_stuff/train_img/000000203744.jpg
diff --git a/datasets/coco_stuff/train_img/000000284465.jpg b/datasets/coco_stuff/train_img/000000284465.jpg
diff --git a/datasets/coco_stuff/train_img/000000350505.jpg b/datasets/coco_stuff/train_img/000000350505.jpg
diff --git a/datasets/coco_stuff/train_img/000000371376.jpg b/datasets/coco_stuff/train_img/000000371376.jpg
diff --git a/datasets/coco_stuff/train_img/000000426773.jpg b/datasets/coco_stuff/train_img/000000426773.jpg
diff --git a/datasets/coco_stuff/train_img/000000475177.jpg b/datasets/coco_stuff/train_img/000000475177.jpg
diff --git a/datasets/coco_stuff/train_img/000000500044.jpg b/datasets/coco_stuff/train_img/000000500044.jpg
diff --git a/datasets/coco_stuff/train_img/000000580986.jpg b/datasets/coco_stuff/train_img/000000580986.jpg
diff --git a/datasets/coco_stuff/train_inst/000000017914.png b/datasets/coco_stuff/train_inst/000000017914.png
diff --git a/datasets/coco_stuff/train_inst/000000029286.png b/datasets/coco_stuff/train_inst/000000029286.png
diff --git a/datasets/coco_stuff/train_inst/000000138805.png b/datasets/coco_stuff/train_inst/000000138805.png
diff --git a/datasets/coco_stuff/train_inst/000000184101.png b/datasets/coco_stuff/train_inst/000000184101.png
diff --git a/datasets/coco_stuff/train_inst/000000197384.png b/datasets/coco_stuff/train_inst/000000197384.png
diff --git a/datasets/coco_stuff/train_inst/000000203744.png b/datasets/coco_stuff/train_inst/000000203744.png
diff --git a/datasets/coco_stuff/train_inst/000000284465.png b/datasets/coco_stuff/train_inst/000000284465.png
diff --git a/datasets/coco_stuff/train_inst/000000350505.png b/datasets/coco_stuff/train_inst/000000350505.png
diff --git a/datasets/coco_stuff/train_inst/000000371376.png b/datasets/coco_stuff/train_inst/000000371376.png
diff --git a/datasets/coco_stuff/train_inst/000000426773.png b/datasets/coco_stuff/train_inst/000000426773.png
diff --git a/datasets/coco_stuff/train_inst/000000475177.png b/datasets/coco_stuff/train_inst/000000475177.png
diff --git a/datasets/coco_stuff/train_inst/000000500044.png b/datasets/coco_stuff/train_inst/000000500044.png
diff --git a/datasets/coco_stuff/train_inst/000000580986.png b/datasets/coco_stuff/train_inst/000000580986.png
diff --git a/datasets/coco_stuff/train_label/000000017914.png b/datasets/coco_stuff/train_label/000000017914.png
diff --git a/datasets/coco_stuff/train_label/000000029286.png b/datasets/coco_stuff/train_label/000000029286.png
diff --git a/datasets/coco_stuff/train_label/000000138805.png b/datasets/coco_stuff/train_label/000000138805.png
diff --git a/datasets/coco_stuff/train_label/000000184101.png b/datasets/coco_stuff/train_label/000000184101.png
diff --git a/datasets/coco_stuff/train_label/000000197384.png b/datasets/coco_stuff/train_label/000000197384.png
diff --git a/datasets/coco_stuff/train_label/000000203744.png b/datasets/coco_stuff/train_label/000000203744.png
diff --git a/datasets/coco_stuff/train_label/000000284465.png b/datasets/coco_stuff/train_label/000000284465.png
diff --git a/datasets/coco_stuff/train_label/000000350505.png b/datasets/coco_stuff/train_label/000000350505.png
diff --git a/datasets/coco_stuff/train_label/000000371376.png b/datasets/coco_stuff/train_label/000000371376.png
diff --git a/datasets/coco_stuff/train_label/000000426773.png b/datasets/coco_stuff/train_label/000000426773.png
diff --git a/datasets/coco_stuff/train_label/000000475177.png b/datasets/coco_stuff/train_label/000000475177.png
diff --git a/datasets/coco_stuff/train_label/000000500044.png b/datasets/coco_stuff/train_label/000000500044.png
diff --git a/datasets/coco_stuff/train_label/000000580986.png b/datasets/coco_stuff/train_label/000000580986.png
diff --git a/datasets/coco_stuff/val_img/000000000139.jpg b/datasets/coco_stuff/val_img/000000000139.jpg
diff --git a/datasets/coco_stuff/val_img/000000000785.jpg b/datasets/coco_stuff/val_img/000000000785.jpg
diff --git a/datasets/coco_stuff/val_img/000000001268.jpg b/datasets/coco_stuff/val_img/000000001268.jpg
diff --git a/datasets/coco_stuff/val_img/000000001490.jpg b/datasets/coco_stuff/val_img/000000001490.jpg
diff --git a/datasets/coco_stuff/val_img/000000001503.jpg b/datasets/coco_stuff/val_img/000000001503.jpg
diff --git a/datasets/coco_stuff/val_img/000000001584.jpg b/datasets/coco_stuff/val_img/000000001584.jpg
diff --git a/datasets/coco_stuff/val_img/000000001818.jpg b/datasets/coco_stuff/val_img/000000001818.jpg
diff --git a/datasets/coco_stuff/val_img/000000001993.jpg b/datasets/coco_stuff/val_img/000000001993.jpg
diff --git a/datasets/coco_stuff/val_inst/000000000139.png b/datasets/coco_stuff/val_inst/000000000139.png
diff --git a/datasets/coco_stuff/val_inst/000000000785.png b/datasets/coco_stuff/val_inst/000000000785.png
diff --git a/datasets/coco_stuff/val_inst/000000001268.png b/datasets/coco_stuff/val_inst/000000001268.png
diff --git a/datasets/coco_stuff/val_inst/000000001490.png b/datasets/coco_stuff/val_inst/000000001490.png
diff --git a/datasets/coco_stuff/val_inst/000000001503.png b/datasets/coco_stuff/val_inst/000000001503.png
diff --git a/datasets/coco_stuff/val_inst/000000001584.png b/datasets/coco_stuff/val_inst/000000001584.png
diff --git a/datasets/coco_stuff/val_inst/000000001818.png b/datasets/coco_stuff/val_inst/000000001818.png
diff --git a/datasets/coco_stuff/val_inst/000000001993.png b/datasets/coco_stuff/val_inst/000000001993.png
diff --git a/datasets/coco_stuff/val_label/000000000139.png b/datasets/coco_stuff/val_label/000000000139.png
diff --git a/datasets/coco_stuff/val_label/000000000785.png b/datasets/coco_stuff/val_label/000000000785.png
diff --git a/datasets/coco_stuff/val_label/000000001268.png b/datasets/coco_stuff/val_label/000000001268.png
diff --git a/datasets/coco_stuff/val_label/000000001490.png b/datasets/coco_stuff/val_label/000000001490.png
diff --git a/datasets/coco_stuff/val_label/000000001503.png b/datasets/coco_stuff/val_label/000000001503.png
diff --git a/datasets/coco_stuff/val_label/000000001584.png b/datasets/coco_stuff/val_label/000000001584.png
diff --git a/datasets/coco_stuff/val_label/000000001818.png b/datasets/coco_stuff/val_label/000000001818.png
diff --git a/datasets/coco_stuff/val_label/000000001993.png b/datasets/coco_stuff/val_label/000000001993.png
diff --git a/docs/images/teaser.jpg b/docs/images/teaser.jpg
diff --git a/gm_Generate New Label.ipynb b/gm_Generate New Label.ipynb
diff --git a/gm_Label Check.ipynb b/gm_Label Check.ipynb
diff --git a/gm_TTTT/test2_img/004.jpg b/gm_TTTT/test2_img/004.jpg
diff --git a/gm_TTTT/test2_label/004.png b/gm_TTTT/test2_label/004.png
diff --git a/gm_TTTT/test_img/001_06.jpg b/gm_TTTT/test_img/001_06.jpg
diff --git a/gm_TTTT/test_label/001.png b/gm_TTTT/test_label/001.png
diff --git a/gm_TTTT/test_label1/001_01.png b/gm_TTTT/test_label1/001_01.png
diff --git a/gm_TTTT/test_label2/001_02.png b/gm_TTTT/test_label2/001_02.png
diff --git a/gm_TTTT/test_label3/001_03.png b/gm_TTTT/test_label3/001_03.png
diff --git a/gm_TTTT/test_label4/001_04.png b/gm_TTTT/test_label4/001_04.png
diff --git a/gm_TTTT/test_label5/001_05.png b/gm_TTTT/test_label5/001_05.png
diff --git a/gm_TTTT/test_label6/001_06.png b/gm_TTTT/test_label6/001_06.png
diff --git a/gm_TTTT/val_img/001.jpg b/gm_TTTT/val_img/001.jpg
diff --git a/gm_TTTT/val_img2/003_03.jpg b/gm_TTTT/val_img2/003_03.jpg
diff --git a/gm_TTTT/val_img3/008_02.jpg b/gm_TTTT/val_img3/008_02.jpg
diff --git a/gm_TTTT/val_label/001.png b/gm_TTTT/val_label/001.png
diff --git a/gm_TTTT/val_label2/003.png b/gm_TTTT/val_label2/003.png
diff --git a/gm_TTTT/val_label2_01/003_01.png b/gm_TTTT/val_label2_01/003_01.png
diff --git a/gm_TTTT/val_label2_02/003_02.png b/gm_TTTT/val_label2_02/003_02.png
diff --git a/gm_TTTT/val_label2_03/003_03.png b/gm_TTTT/val_label2_03/003_03.png
diff --git a/gm_TTTT/val_label3/008.png b/gm_TTTT/val_label3/008.png
diff --git a/gm_TTTT/val_label3_01/008_01.png b/gm_TTTT/val_label3_01/008_01.png
diff --git a/gm_TTTT/val_label3_02/008_02.png b/gm_TTTT/val_label3_02/008_02.png
diff --git a/gm_TestTimeTrainingTest.ipynb b/gm_TestTimeTrainingTest.ipynb
diff --git a/gm_Visualize.ipynb b/gm_Visualize.ipynb
diff --git a/gm_test.py b/gm_test.py
@@ -0,0 +1,47 @@
+"""
+Copyright (C) 2019 NVIDIA Corporation.  All rights reserved.
+Licensed under the CC BY-NC-SA 4.0 license (https://creativecommons.org/licenses/by-nc-sa/4.0/legalcode).
+"""
+
+import os
+from collections import OrderedDict
+
+import data
+from options.test_options import TestOptions
+from models.pix2pix_model import Pix2PixModel
+from util.visualizer import Visualizer
+from util import html
+
+opt = TestOptions().parse()
+
+dataloader = data.create_dataloader(opt)
+
+model = Pix2PixModel(opt)
+model.eval()
+
+visualizer = Visualizer(opt)
+
+# create a webpage that summarizes the all results
+# web_dir = os.path.join(opt.results_dir, opt.name,
+#                        '%s_%s' % (opt.phase, opt.which_epoch))
+
+
+webpage = html.HTML(web_dir,
+                    'Experiment = %s, Phase = %s, Epoch = %s' %
+                    (opt.name, opt.phase, opt.which_epoch))
+
+# test
+for i, data_i in enumerate(dataloader):
+    if i * opt.batchSize >= opt.how_many:
+        break
+
+    generated = model(data_i, mode='inference')
+
+    img_path = data_i['path']
+    for b in range(generated.shape[0]):
+        print('process image... %s' % img_path[b])
+        visuals = OrderedDict([('input_label', data_i['label'][b]),
+                               ('synthesized_image', generated[b])])
+        visualizer.save_images(webpage, visuals, img_path[b:b + 1])
+
+webpage.save()
diff --git a/models/__init__.py b/models/__init__.py
diff --git a/models/networks/Synchronized-BatchNorm-PyTorch b/models/networks/Synchronized-BatchNorm-PyTorch
diff --git a/models/networks/sync_batchnorm/__init__.py b/models/networks/sync_batchnorm/__init__.py
@@ -0,0 +1,14 @@
+# -*- coding: utf-8 -*-
+# File   : __init__.py
+# Author : Jiayuan Mao
+# Email  : [email protected]
+# Date   : 27/01/2018
+#
+# This file is part of Synchronized-BatchNorm-PyTorch.
+# https://github.com/vacancy/Synchronized-BatchNorm-PyTorch
+# Distributed under MIT License.
+
+from .batchnorm import set_sbn_eps_mode
+from .batchnorm import SynchronizedBatchNorm1d, SynchronizedBatchNorm2d, SynchronizedBatchNorm3d
+from .batchnorm import patch_sync_batchnorm, convert_model
+from .replicate import DataParallelWithCallback, patch_replication_callback