It shows a red crab on a beach in front of the bright blue ocean with a blue sky and white clouds.

I guess transfer of the wooden house amidst yellow fields with a reddening sky might lead to a wooden crab on a yellow field in front of a reddish-yellow ocean with red sky and clouds, or something.

It actually looks better than expected: https://imgur.com/a/5BjvC

Looks nice!

Did you have to do anything extra to get it working? I've set things up according to the documentation (I think), but I get dimension size errors when running it.

Haha, yes, I had to rewrite their code a bit. All the .unsqueeze(1).expand_as(...) in photo_wct.py need to be replaced by just .expand_as(...) and the return value of __feature_wct needs to be wrapped in torch.autograd.Variable.

I'm going to submit a PR, but it took me a bit of experimentation to fix these errors, so the code is a bit messier than I'd like.

Ahh that looks like the error I was hitting, thanks. I might try replacing the bits as well, though I just upgraded pytorch from 0.1.12 to 0.3 and it became much slower (I killed it after 5-6 minutes of setup).

My fork is here: https://github.com/Yorwba/FastPhotoStyle

I was using the pytorch 0.1.12 installed with conda (following their USAGE.md) and it took ~30s total for the transfer.

Much appreciated thanks!

For some reason it's taking me about 4-5 minutes for the transfer, but the code now runs and the rest of the runtime is only a few seconds.

Wow, thats pretty good! So this thing can do fairly well on complex transfers.