Depth head facelift - Githubissues

cc-ai / climategan

Code and pre-trained model for the algorithm generating visualisations of 3 climate change related events: floods, wildfires and smog.

https://thisclimatedoesnotexist.com

GNU General Public License v3.0

75 stars 18 forks source link

Closed melisandeteng closed 4 years ago

melisandeteng commented 4 years ago

In this PR, the following changes are made to the depth head:

[x] input depth is a npy array of log depth
[x] log inverse depth images in comet
[x] change loss to scale invariant MSE loss
[x] change loader and trasforms so that they can be applied to float tensors
[x] change image logging if you want to check what the inputs would look like, npy arrays of log depth predictions from Megadepth have been saved at /network/tmp1/ccai/data/munit_dataset/trainA_megadepth_resized

melisandeteng commented 4 years ago

@vict0rsch @alexrey88 @51N84D @tianyu-z @sashavor can you please review this ? Thanks !