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
76 stars 18 forks source link

generator architectures => Semantic Segmentation Head #10

Closed vict0rsch closed 3 years ago

vict0rsch commented 4 years ago

create UNet structure

vict0rsch commented 4 years ago

work in progress in #14

vict0rsch commented 4 years ago

14 merged but:

@adrienju could you merge your work from MUNIT into Omnigan? Meaning, integrating a pretrained (and trainable) segmentation head? Keep in mind: you'll also have to adjust losses, tests and possibly data-loading :) Happy coding !