TensorFlow
Darknet
YOLOv2
darkflow

darkflowでAssertionErrorが出た場合の対処

tensorflow版のyolov2:https://github.com/thtrieu/darkflowにて
本家のYOLOで学習させたmodeとcfgを用いて、predictをやろうとすると以下のエラーが出た。

AssertionError: expect 202314760 bytes, found 202314764

↑4byteだけ違う。
4byteどころではなく明らかに数値が違った場合はおそらくtrain時のcfgとpredict時のcfgが異なっている

対処方法は、darkflow/darkflow/utils/loader.pyの121行目

loader.py(121行目)
self.offset = 16

となっているところを

self.offset = 20

に書き換えれば直る。

参考