前言
代码已经在Github开源:BshoterJ ,也可以查看我朋友唐博士的:INFINITSY
demo:
简介
项目介绍
这个项目来自本科实验课设,当初的选题是显著性检测下的图像检索,类似于淘宝的拍立得那种功能。我们在狗年结束之际打算做一个“搜狗”的脚本,识别品种并将与之最相近的若干张图库中的图输出。不过既然不是直接提特征检索,那么我们可以选目标较多比较难分的数据集了。
实验需要
实验平台:ubuntu14.06 + Python3.6 + Pytorch0.4 + keras
数据集:来自Stanford Dogs Dataset
我们选取其中较常见的18种狗,由于每种狗只提供150-200张左右,我们分了训练集2533张,验证集629张,图库18类*25张。