FashionAI2018 服装属性标签识别
环境
caffe+keras
python 2.7
numpy 1.14.2
opencv 3.4.0
cuda9.0
cudnn 7.0
文件说明:
caffe部分:
使用InceptionV4举例,该网络在复赛中可以实现94.11的准确率。
训练网络使用training.sh
修改相应的路径,即Log、TOOLS、-weights后面的路径,其中,Inception-V4的pretrain-model请到下载 。
使用方法bash training.sh [solver][Class][GPU]。
solver : solver文件路径,本代码中存放于solver文件夹
class: 类别名,用来命名log日志文件,即neck,collar...
GPU : GPU的型号,如0,1,2
2. caffe源码处理
,并用文件中的src和inc
1