之前封装的代码只能有一个输出:http://blog.youkuaiyun.com/sinat_30071459/article/details/51823390
并且有点问题,现在作了修改,模型可以有多个输出(当然也可以只有一个输出),可用于做图片的多属性分类,如:
这个例子有四个输出;
代码输出的分类结果为:
int为标签下标(0,1,2,3...),float为确信度。predictions为模型所有输出的结果,每个结果取确信度排前二的情况。
如果想显示具体的标签,可自行修改代码。
代码下载地址:http://download.youkuaiyun.com/detail/sinat_30071459/9715053