前言
最近在做人脸关键点检测,已经用pytorch跑起来了,因为之前没学过caffe,故这次打算用caffe做,谁知用caffe做数据不好标注
caffe数据来源与标注
caffe的数据可以来自于:
- 数据库(如 LevelDB 和 LMDB)
- 内存(MemoryData)
- HDF5(HDF5Data)
- 图片(ImageData)
- Windows(WindowData)
其中caffe的标注过程都需要txt文件,其中每行左侧为文件名(可带路径),之后加一个空格,然后是标注,这个标注是离散的,故常用于分类任务,在其他地方看到使用xml文件而用于检测任务,而做回归的比较少,不过后面还是发现了一些讨论:
https://www.cnblogs.com/frombeijingwithlove/p/5314042.html
https://www.zhihu.com/question/37351143
从讨论可以看出caffe做回归任务尤其是关键点检测这种向量回归任务不太适合,需要大量的其他工作。
所以,等caffe2成熟了适合部署再去吧,咱现在还是用tensorflow吧。
发现用HDF5可以做向量回归,但是相对较麻烦。