search

friedhelm739 / Insightface-tensorflow

Face Recognization System By MTCNN & Insightface
46 stars 18 forks source link

大佬，怎么感觉gen_tfrecord_mxdata.py运行不通啊 #4

Open zemelLeong opened 5 years ago

zemelLeong commented 5 years ago

s = imgrec.read_idx(0) 报错：mxnet.base.MXNetError: [15:19:57] C:\Jenkins\workspace\mxnet-tag\mxnet\3rdparty\dmlc-core\src\recordio.cc:65: Check failed: header[0] == RecordIOWriter::kMagic: Invalid RecordIO File
如果把0改为1的话就正常
imgidx = list(range(1, int(header.label[0])))
报错：TypeError: 'float' object is not subscriptable
header.label得到的是一个浮点数

friedhelm739 commented 5 years ago

我这里运行是没有问题的呀 你可以去下面的链接去看看，这块代码我是参考的这个链接 https://github.com/luckycallor/InsightFace-tensorflow

zemelLeong commented 5 years ago

我稍微改了下代码能运行了
我配置比较低，跑了一天了loss 在17左右，在训练集上的精度在0到0.6之间
我使用的batch_size为10，次数达到了374000 请问你当时跑了多久啊？loss到多低才可用呢