Comment créer un fichier HDF5 (classification des étiquettes mutli) sur fichier txt à utiliser dans Caffe

Question

je la structure suivante dans un fichier txt:

/path/to/image x y 
/path/to/image x y 

où x et y sont des nombres entiers.

Ce que je veux faire est maintenant: Créez un fichier hdf5 à utiliser dans Caffe ('train.prototxt')

code Python Mon ressemble à ceci:

import h5py 
import numpy as np 
import os 

text = 'train' 
text_dir = text + '.txt' 

data = np.genfromtxt(text_dir, delimiter=" ", dtype=None) 

h = h5py.File(text + '.hdf5', 'w') 
h.create_dataset('data', data=data[:1]) 
h.create_dataset('label', data=data[1:]) 

with open(text + "_hdf5.txt", "w") as textfile: 
    textfile.write(os.getcwd() + '/' +text + '.hdf5') 

Mais cela ne fonctionne pas! Des idées sur ce qui pourrait être faux?

Answer 1

Cela ne fonctionne pas car votre 'data' est /path/to/image au lieu de l'image elle-même. Pour plus d'informations, voir this answer.