search

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

add Depth head #71

Closed melisandeteng closed 4 years ago

melisandeteng commented 4 years ago

Add depth auxiliary task. json files with depth have yet to be generated but this is functional

Raw input data : images Depth in dataloader item: tensor We read the input images with the pil_image_loader function to apply the same transforms over all inputs (the original image, seg, mask ) but then consider that depth map as a tensor.

vict0rsch commented 4 years ago

@alexrey88 @51N84D @tianyu-z @sashavor can you have a look at the PR?