关于caffe做向量回归的问题

前言

最近在做人脸关键点检测,已经用pytorch跑起来了,因为之前没学过caffe,故这次打算用caffe做,谁知用caffe做数据不好标注

caffe数据来源与标注

caffe的数据可以来自于:

  1. 数据库(如 LevelDB 和 LMDB)
  2. 内存(MemoryData)
  3. HDF5(HDF5Data)
  4. 图片(ImageData)
  5. Windows(WindowData)

其中caffe的标注过程都需要txt文件,其中每行左侧为文件名(可带路径),之后加一个空格,然后是标注,这个标注是离散的,故常用于分类任务,在其他地方看到使用xml文件而用于检测任务,而做回归的比较少,不过后面还是发现了一些讨论:
https://www.cnblogs.com/frombeijingwithlove/p/5314042.html
https://www.zhihu.com/question/37351143
从讨论可以看出caffe做回归任务尤其是关键点检测这种向量回归任务不太适合,需要大量的其他工作。
所以,等caffe2成熟了适合部署再去吧,咱现在还是用tensorflow吧。

发现用HDF5可以做向量回归,但是相对较麻烦。

你可能感兴趣的:(caffe)