VGG猫狗大战

 

 

 

 

 

 

 

 

 

 

 

 首先将数据进行预处理，取中心大小224*224，转化成tensor,并进行标准化，并将数据放到队列当中

设置VGG的最后全连接层为（4096，2），分两类最后用逻辑回归的softmax计算出该网络将该图片分到猫或狗哪类的概率

posted @ 2020-07-29 16:59  a张南a  阅读(180)  评论(0)    收藏  举报