对抗网络GAN生成图片之微信插件实现

对抗网络GAN生成图片之微信插件实现

大家好，我是奇点。

本周在朋友圈里，有一个叫云萌宠的小产品，在微信公众号里回复”撸猫“或”撸狗“，微信公众号就会回复一张机器生成”不存在“的图片。上线不久，直接把服务器给撸挂了。接下来，我们讲一下微信插件的代码实现。

本插件是基于于开源建站工具JPress开发的，JPress的插件可以在线安装或卸载，使用起来非常方便。接下来，我们直接进入正题，代码集成过程。

核心代码

如果还对GAN或云萌宠不太了解的读者，可以简单回顾一下奇点之前写的文章，在这里不再展开讲解。为了更好地集成和使用，我们对之前的代码进行简单地改造，分成加载模型和预测两部分。具体代码如下：

public class GanKit {
   
	private static Log log = Log.getLog(GanKit.class);

	public static GanKit me = new GanKit();

	private Criteria<int[], Image[]> criteria;
	private ZooModel<int[], Image[]> model;

	private GanKit() {
   
	}

	public synchronized Criteria<int[], Image[]> gan() {
   
		if (criteria == null) {
   
			criteria = Criteria.builder().
					optApplication(Application.CV.IMAGE_GENERATION)
					.setTypes(int[].class, Image[].class)
					.optFilter("size", "256")
					.optArgument("truncation"