10 from collections
import defaultdict
12 if __name__ ==
"__main__":
13 parser = argparse.ArgumentParser()
14 parser.add_argument(
'--leveldb', type=str, required=
True)
16 args = parser.parse_args()
20 matrix= np.zeros(392, dtype =
'f4' )
23 db = leveldb.LevelDB(args.leveldb)
26 for key, value
in db.RangeIter():
30 datum = caffe.proto.caffe_pb2.Datum()
31 datum.ParseFromString(value)
32 label =
int(datum.label)
39 for i
in range(0, 392):
40 if matrix[i] < (0.001*count):
41 matrix[i]= (0.001*count)
44 total=total+matrix[i]*((1./(matrix[i]/count))/(3920.*count))
45 matrix[i]=(1./(matrix[i]/count))/(3920.)
47 matrix[i]=0.7+(matrix[i])
55 summatrix=np.zeros((392, 392), dtype =
'f4')
56 for i
in range(0, 392):
57 summatrix[i][i]=matrix[i]
65 blob=caffe.io.array_to_blobproto(readyForH.reshape(1,1,392,392))
67 with
open(
'infoGain.binaryproto',
'wb')
as f:
68 f.write(blob.SerializeToString())
procfile open("FD_BRL_v0.txt")