search

CSAILVision / gandissect

Pytorch-based tools for visualizing and understanding the neurons of a GAN. https://gandissect.csail.mit.edu/
MIT License
1.77k stars 282 forks source link

GanImageSegmenter #7

Closed alexzhou907 closed 5 years ago

alexzhou907 commented 5 years ago

Is there any code provided for GanImageSegmenter as used in ablate.py?

davidbau commented 5 years ago

Cleaned up a couple issues.

  1. evalablate.py is the new version of the file ablate.py, so deleted ablate.py.
  2. UnifiedParsingSegmenter is the new image segmenter used, so mentioned that in the comments.
  3. Updated code for the unified parsing segmenter code so that it can be used on pytorch 1.0+

Fiexd at a529e3cc64e5c9553c802002782351088c892eaf