Skip to content

Commit

Permalink
add map2sat & update test script
Browse files Browse the repository at this point in the history
  • Loading branch information
junyanz committed Jan 31, 2019
1 parent a0f535f commit 28dc61b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions scripts/download_pix2pix_model.sh
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
FILE=$1

echo "Note: available models are edges2shoes, sat2map, facades_label2photo, and day2night"

echo "Note: available models are edges2shoes, sat2map, map2sat, facades_label2photo, and day2night"
echo "Specified [$FILE]"

mkdir -p ./checkpoints/${FILE}_pretrained
Expand Down
2 changes: 1 addition & 1 deletion test.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
if __name__ == '__main__':
opt = TestOptions().parse() # get test options
# hard-code some parameters for test
opt.num_threads = 1 # test code only supports num_threads = 1
opt.num_threads = 0 # test code only supports num_threads = 1
opt.batch_size = 1 # test code only supports batch_size = 1
opt.serial_batches = True # disable data shuffling; comment this line if results on randomly chosen images are needed.
opt.no_flip = True # no flip; comment this line if results on flipped images are needed.
Expand Down

0 comments on commit 28dc61b

Please sign in to comment.