diff --git a/models/megengine_vision_gan.md b/models/megengine_vision_gan.md index e392f48102bdaac44e1baa40c9c3f573e0d73880..075625d917af3427751b01a441472f74a33ed2ab 100644 --- a/models/megengine_vision_gan.md +++ b/models/megengine_vision_gan.md @@ -9,6 +9,18 @@ tags: [vision, gan] github-link: https://github.com/MegEngine/Models/tree/master/official/vision/gan --- +```python3 +import megengine.hub as hub +import megengine_mimicry.nets.dcgan.dcgan_cifar as dcgan +import megengine_mimicry.utils.vis as vis + +netG = dcgan.DCGANGeneratorCIFAR() +netG.load_state_dict(hub.load_serialized_obj_from_url("https://data.megengine.org.cn/models/weights/dcgan_cifar.pkl")) +images = dcgan_generator.generate_images(num_images=64) # in NCHW format with normalized pixel values in [0, 1] +grid = vis.make_grid(images) # in HW3 format with [0, 255] BGR images for visualization +vis.save_image(grid, "visual.png") +``` + #### 训练参数