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

Infer segmentation #153

Closed alexrey88 closed 3 years ago

alexrey88 commented 3 years ago

Modifying infer.py to also be able to infer the segmentation tensors from images