def testMix(self):
acmodel1 = AcModel()
hmm1 = HMM()
hmm1.create_proto( 25 )
hmm1.name = "y"
acmodel1.append_hmm( hmm1 )
acmodel1.repllist.add("y","j")
acmodel2 = AcModel()
hmm2 = HMM()
hmm2.create_proto( 25 )
hmm2.name = "j"
hmm3 = HMM()
hmm3.create_proto( 25 )
hmm3.name = "y"
acmodel2.hmms.append( hmm2 )
acmodel2.hmms.append( hmm3 )
acmodel2.repllist.add("y","y")
acmodel2.repllist.add("j","j")
modelmixer = ModelMixer()
modelmixer.set_models( acmodel1,acmodel2 )
outputdir = os.path.join(MODELDIR, "models-test")
modelmixer.mix( outputdir, gamma=1. )
mixedh1 = AcModel()
mixedh1.load( outputdir )
shutil.rmtree( outputdir )
class TestAcModel(unittest.TestCase):
#This one takes too much time to be tested each time....
# def test_load_all_models(self):
# models = glob.glob(os.path.join(MODEL_PATH,"models-*","hmmdefs"))
# for hmmdefs in models:
# acmodel = AcModel( hmmdefs )
# self._test_load_save( acmodel )
def setUp(self):
self.hmmdefs = os.path.join(MODEL_PATH,"models-jpn","hmmdefs")
self.acmodel = AcModel()
self.acmodel.load_htk( self.hmmdefs )
def test_load_save(self):
self._test_load_save( self.hmmdefs )
def _test_load_save(self, acmodel):
# Save temporary the loaded model into a file
tmpfile = self.hmmdefs+".copy"
self.acmodel.save_htk( tmpfile )
# Load the temporary file into a new model
acmodelcopy = AcModel()
acmodelcopy.load_htk(tmpfile)
# Compare original and copy
self.assertEqual(len(self.acmodel.hmms),len(acmodelcopy.hmms))
for hmm,hmmcopy in zip(self.acmodel.hmms,acmodelcopy.hmms):
self.assertEqual(hmm.name,hmmcopy.name)
self.assertTrue(compare(hmm.definition,hmmcopy.definition))
self.assertTrue(compare(self.acmodel.macros,acmodelcopy.macros))
os.remove(tmpfile)
def test_get_hmm(self):
with self.assertRaises(ValueError):
self.acmodel.get_hmm('Q')
Nhmm = self.acmodel.get_hmm('N')
self.__test_states( Nhmm.definition['states'] )
self.__test_transition( Nhmm.definition['transition'] )
def test_append_hmm(self):
with self.assertRaises(TypeError):
self.acmodel.append_hmm({'toto':None})
hmm = HMM()
with self.assertRaises(TypeError):
self.acmodel.append_hmm(hmm)
Nhmm = self.acmodel.get_hmm('N')
with self.assertRaises(ValueError):
self.acmodel.append_hmm(Nhmm)
Newhmm = copy.deepcopy(Nhmm)
Newhmm.name = "NewN"
self.acmodel.append_hmm(Newhmm)
def test_pop_hmm(self):
self.acmodel.pop_hmm("N")
with self.assertRaises(ValueError):
self.acmodel.get_hmm( "N" )
def test_load_hmm(self):
hmm = HMM()
hmm.load( os.path.join(HERE,"N-hmm") )
self.__test_states( hmm.definition['states'] )
self.__test_transition( hmm.definition['transition'] )
def test_save_hmm(self):
hmm = HMM()
hmm.load( os.path.join(HERE,"N-hmm") )
hmm.save(os.path.join(HERE,"N-hmm-copy"))
newhmm = HMM()
newhmm.load(os.path.join(HERE,"N-hmm-copy"))
os.remove(os.path.join(HERE,'N-hmm-copy'))
self.assertEqual(hmm.name,newhmm.name)
self.assertTrue(compare(hmm.definition,newhmm.definition))
def test_fill(self):
acmodel1 = AcModel()
acmodel1.load_htk( os.path.join(HERE,"1-hmmdefs") )
ahmm1=acmodel1.get_hmm('a')
a1transition = [macro["transition"] for macro in acmodel1.macros if macro.get('transition',None)][0]
acmodel1.fill_hmms()
self.assertTrue(compare(ahmm1.definition['transition'],a1transition['definition']))
def test_no_merge(self):
nbhmms = len(self.acmodel.hmms)
# Try to merge with the same model!
acmodel2 = AcModel()
acmodel2.load_htk(self.hmmdefs)
(appended,interpolated,keeped,changed) = acmodel2.merge_model(self.acmodel,gamma=1.)
self.assertEqual(interpolated, 0)
self.assertEqual(appended, 0)
self.assertEqual(keeped, nbhmms)
self.assertEqual(changed, 0)
(appended,interpolated,keeped,changed) = acmodel2.merge_model(self.acmodel,gamma=0.5)
self.assertEqual(interpolated, nbhmms)
self.assertEqual(appended, 0)
self.assertEqual(keeped, 0)
self.assertEqual(changed, 0)
(appended,interpolated,keeped,changed) = acmodel2.merge_model(self.acmodel,gamma=0.)
self.assertEqual(interpolated, 0)
self.assertEqual(appended, 0)
self.assertEqual(keeped, 0)
self.assertEqual(changed, nbhmms)
# Try to merge with a different MFCC parameter kind model...
acmodel2 = AcModel()
acmodel2.load_htk( os.path.join(MODEL_PATH,"models-cat","hmmdefs") )
with self.assertRaises(TypeError):
acmodel2.merge_model(self.acmodel,gamma=1.)
def test_merge(self):
acmodel1 = AcModel()
acmodel1.load_htk( os.path.join(HERE,"1-hmmdefs") )
acmodel2 = AcModel()
acmodel2.load_htk( os.path.join(HERE,"2-hmmdefs") )
(appended,interpolated,keeped,changed) = acmodel2.merge_model(acmodel1,gamma=0.5)
self.assertEqual(interpolated, 2) # acopy, a
self.assertEqual(appended, 1) # i
self.assertEqual(keeped, 1) # e
self.assertEqual(changed, 0)
self.__test_states( acmodel2.get_hmm('a').definition['states'] )
self.__test_transition( acmodel2.get_hmm('a').definition['transition'] )
def test_replace_phones(self):
acmodel1 = AcModel()
acmodel1.load( os.path.join(MODEL_PATH,"models-fra") )
acmodel1.replace_phones( reverse=False )
acmodel1.replace_phones( reverse=True )
acmodel2 = AcModel()
acmodel2.load( os.path.join(MODEL_PATH,"models-fra") )
for h1 in acmodel1.hmms:
h2 = acmodel2.get_hmm( h1.name )
self.assertTrue(compare(h1.definition['transition'],h2.definition['transition']))
self.assertTrue(compare(h1.definition['states'],h2.definition['states']))
def test_monophones(self):
acmodel1 = AcModel()
acmodel1.load( os.path.join(MODEL_PATH,"models-fra") )
acmodel2 = acmodel1.extract_monophones()
acmodel2.save(os.path.join(HERE,'fra-mono'))
self.assertTrue( os.path.isfile( os.path.join(HERE, 'fra-mono','hmmdefs')) )
self.assertTrue( os.path.isfile( os.path.join(HERE, 'fra-mono','monophones.repl')) )
self.assertFalse( os.path.isfile( os.path.join(HERE, 'fra-mono','tiedlist')) )
os.remove( os.path.join(HERE, 'fra-mono','hmmdefs') )
os.remove( os.path.join(HERE, 'fra-mono','monophones.repl') )
os.rmdir( os.path.join(HERE,'fra-mono') )
self.assertEqual( len(acmodel2.hmms), 38 )
def test_proto(self):
h1 = HtkIO()
h1.write_hmm_proto( 25, os.path.join(HERE,"proto_from_htkio") )
h2 = HMM()
h2.create_proto( 25 )
h2.save( os.path.join(HERE,"proto_from_hmm") )
m1 = HMM()
m1.load( os.path.join(HERE,"proto_from_htkio") )
m2 = HMM()
m2.load( os.path.join(HERE,"proto_from_hmm") )
self.assertTrue(compare(m1.definition['transition'],m2.definition['transition']))
self.assertTrue(compare(m1.definition['states'],m2.definition['states']))
os.remove( os.path.join(HERE,"proto_from_hmm") )
os.remove( os.path.join(HERE,"proto_from_htkio") )
def __test_transition(self, transition):
self.assertEqual(transition['dim'], 5)
matrix = transition['matrix']
for i in range(len(matrix)-1):
# the last vector is always 0.!
vector=matrix[i]
self.assertEqual(1.0, round(sum(vector),4))
def __test_states(self, states):
for item in states: # a dict
state = item['state']
streams = state['streams']
for s in streams: # a list
mixtures = s['mixtures']
for mixture in mixtures: # a list of dict
#self.assertEqual(type(mixture['weight']),float)
pdf = mixture['pdf']
self.assertEqual(pdf['mean']['dim'], 25)
self.assertEqual(len(pdf['mean']['vector']),25)
self.assertEqual(pdf['covariance']['variance']['dim'], 25)
self.assertEqual(len(pdf['covariance']['variance']['vector']), 25)
self.assertEqual(type(pdf['gconst']),float)