eval

from gala import evaluate
from gala import imio

import matplotlib.pyplot as plt

# fn = '/data/hanslovskyp/cremi/gala-from-juan/sample_B_20160630_frags.hdf'
fn = '/data/hanslovskyp/cremi/gala/sample_B_20160708_frags_46_55.hdf'
# fn = '/data/hanslovskyp/cremi/gala-from-juan/sample_B_20160501.hdf'
# fn = '/data/hanslovskyp/cremi/gala-from-juan/sample_B_20160630_frags.hdf'
fn = '/data/hanslovskyp/cremi/gala-from-juan/sample_B_20160708_frags_46_55.hdf'

fragments = imio.read_cremi(fn, ['/volumes/labels/fragments'])[0]
gt = imio.read_cremi(fn, ['/volumes/labels/neuron_ids'])[0]
perf = [evaluate.split_vi(f, gt) for (f, gt) in zip(fragments, gt)]
fm, fs = zip(*perf)
X = tuple(range(fragments.shape[0]))
plt.subplot(2,1,1)
plt.legend(handles=plt.plot(X, fm, 'ko', alpha=0.5, label='False Merge'))
plt.subplot(2,1,2)
plt.legend(handles=plt.plot(X, fs, 'ko', alpha=0.5, label='False Split'))

print(fm)
print(fs)


plt.show()