在特定MS COCO类别上微调深度CNN模型
计算机视觉与模式识别
2017-09-07 v1 人工智能
机器学习
摘要
深度卷积神经网络（CNN）的微调通常是需要的。本文概述了我们公开可用的 py-faster-rcnn-ft 软件库，该库可用于在 Microsoft Common Objects in Context (MS COCO) 数据集的自定义子集上微调 VGG_CNN_M_1024 模型。例如，我们改进了流程，使用户不必手动在数据集中寻找合适的图像文件，这些文件随后可用于演示程序。我们的实现随机选取包含至少一个所微调类别物体的图像。
引用
@article{arxiv.1709.01476,
title = {Fine-tuning deep CNN models on specific MS COCO categories},
author = {Daniel Sonntag and Michael Barz and Jan Zacharias and Sven Stauden and Vahid Rahmani and Áron Fóthi and András Lőrincz},
journal= {arXiv preprint arXiv:1709.01476},
year = {2017}
}