Caffe Python Layer für die Eingabe, setzen Sie den Datentyp auf Uint8

Question

Ich versuche, ein CNN in Caffe zu trainieren. Ich wollte eine Menge Daten vergrößern, also verwende ich eine "Python" Ebene für die Eingabe, wie vorgeschlagen here.

Allerdings sehe ich aus dem Protokoll, dass Caffe den Datentyp float32 für alle meine Daten verwendet. Das ist wirklich verschwenderisch, weil ich es nur mit 8-Bit-Ganzzahlen zu tun habe. Gibt es eine Möglichkeit, Caffe zu sagen, dtype='uint8' zu verwenden?

Ich habe versucht, die Daten typisieren, während die Top-Einstellung:

top[0].data[...] = someArray.astype(np.uint8, copy=False) 

aber das funktioniert nicht.

Irgendwelche Vorschläge?

Answer 1

AFAIK, caffe wird derzeit kompiliert, um nur float32 oder float64 zu unterstützen. Ich nehme an, lmdb/leveldb Daten können in uint8 Format gespeichert werden, aber Caffe konvertiert es intern zu float32 beim Lesen.
Die Tatsache, dass Ihre Eingabedaten uint8 sind, bedeutet nicht, dass die gesamte Verarbeitung auf diese Weise erhalten bleibt. Bei der ersten Faltung/InnerProduct-Ebene werden Daten mit Gleitkommazahlen multipliziert und können nicht länger uint8 bleiben.
Also, ich nehme an, Sie sollten mit der kleinen Raumverschwendung auf der Eingabeschicht dulden und die Konvertierung auf uint8 aufgeben.