fix(dataset): set correct output dir path for gen dataset

Author: nicholaslck

.gitignore

@@ -0,0 +1,10 @@
+**/.DS_Store
+**/._.DS_Store
+checkpoints/
+**/__pycache__/**
+
+
+# nicholas local dir
+**/assets/**
+**/out_single/**
+dataset/HalftoneVOC2012/tmp/*

README.md

@@ -56,9 +56,9 @@ python train2_stage1.py -r <path to checkpoint>
 
 # Inference
 python inference_single.py \
-    --input <path to image (.png)> \
-    --out_dir <output directory> \
-    --method 'ours'
+--input [path to image (.png)] \
+--out_dir [output directory] \
+--method 'ours'
 ```
 
 ## Citation

configs/train2_stage1.yml

@@ -11,7 +11,7 @@ training:
 model:
   init_checkpoint:
   reshalf_pretrained: checkpoints/pretrained/reshalf_model_warm.pth.tar
-  invhalf_pretrained: output/train/PRLNet/b/2022-09-30_17:48:03/checkpoints/model_epoch127.pth.tar
+  invhalf_pretrained: checkpoints/pretrained/prlnet_epoch127.tar
   use_input_y: false
   noise_weight: 0.3 # default 0.3
 optimizer:

dataset/HalftoneVOC2012/genImageSet.py

@@ -1,3 +1,4 @@
+import cv2
 import mmcv
 import os
 import numpy as np
@@ -6,9 +7,12 @@
 import subprocess
 
 # Path to original VOC2012 dataset
-input_dir = "dataset/voc2012/VOCdevkit/VOC2012/JPEGImages"
+input_dir = "/research/d4/gds/cklau21/Datasets/voc2012/VOCdevkit/VOC2012/JPEGImages"
 
-output_root_dir = 'HalftoneVOC2012_ov'
+if not os.path.exists(input_dir):
+  raise RuntimeError(f"Cannot find input directory {input_dir}, please update the variable input_dir to point to the JPEGImages directory of VOC2012")
+
+output_root_dir = 'HalftoneVOC2012'
 input_files = [ f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f)) ]
 
 training_set_boundary = 13758
@@ -27,7 +31,7 @@
 
 def dither_and_save(i):
   input_file = input_files[i]
-  raw_input_img = mmcv.imread(os.path.join(input_dir, input_file), 'color')
+  raw_input_img = cv2.imread(os.path.join(input_dir, input_file), cv2.IMREAD_COLOR)
 
   # Crop and resize image into 256x256
   height, width, channel = raw_input_img.shape

dataset/HalftoneVOC2012/genSpecialSet.py

@@ -7,7 +7,7 @@
 import argparse
 import subprocess
 
-output_root_dir = 'special_ov'
+output_root_dir = 'special'
 
 max_set = 13758 + 3367
 training_set_boundary = 13758

inference_single.py

@@ -2,7 +2,6 @@
 import argparse
 import os
 import os.path as path
-from logging import info as log
 from torchvision.transforms import GaussianBlur
 
 import cv2