Skip to content
Snippets Groups Projects
Commit 6a27753f authored by akomaty@idiap.ch's avatar akomaty@idiap.ch
Browse files

added new modules to config files and setup.py

parent 35aea91c
No related branches found
No related tags found
1 merge request!7Diarization
import bob.bio.gmm
algorithm = bob.bio.gmm.algorithm.GMMSegment(
number_of_gaussians = 512,
features_per_seg = 250, # number of features per segment, the default value is 250, which means that if the window shift of mfcc is 10 ms, then the duration of the segment is 2.5 s
seg_overlap = 50, # number of overlapped segments
)
\ No newline at end of file
import bob.bio.gmm
algorithm = bob.bio.gmm.algorithm.IVectorSegment(
# IVector parameters
subspace_dimension_of_t = 100,
update_sigma = True,
tv_training_iterations = 3, # Number of EM iterations for the TV training
# GMM parameters
number_of_gaussians = 512,
features_per_seg = 250, # number of features per segment, the default value is 250, which means that if the window shift of mfcc is 10 ms, then the duration of the segment is 2.5 s
seg_overlap = 50, # number of overlapped segments
)
...@@ -587,7 +587,7 @@ def test_ivector_lda_wccn_plda(): ...@@ -587,7 +587,7 @@ def test_ivector_lda_wccn_plda():
def test_ivector_segment(): def test_ivector_segment():
temp_file = bob.io.base.test_utils.temporary_filename() temp_file = bob.io.base.test_utils.temporary_filename()
ivec1 = bob.bio.base.load_resource("ivec-segment", "algorithm", preferred_package='bob.bio.gmm') ivec1 = bob.bio.base.load_resource("ivector-segment", "algorithm", preferred_package='bob.bio.gmm')
assert isinstance(ivec1, bob.bio.gmm.algorithm.IVectorSegment) assert isinstance(ivec1, bob.bio.gmm.algorithm.IVectorSegment)
assert isinstance(ivec1, bob.bio.gmm.algorithm.GMMSegment) assert isinstance(ivec1, bob.bio.gmm.algorithm.GMMSegment)
assert isinstance(ivec1, bob.bio.base.algorithm.Algorithm) assert isinstance(ivec1, bob.bio.base.algorithm.Algorithm)
......
...@@ -121,12 +121,14 @@ setup( ...@@ -121,12 +121,14 @@ setup(
'bob.bio.algorithm': [ 'bob.bio.algorithm': [
'gmm = bob.bio.gmm.config.algorithm.gmm:algorithm', 'gmm = bob.bio.gmm.config.algorithm.gmm:algorithm',
'gmm-segment = bob.bio.gmm.config.algorithm.gmm_segment:algorithm', # GMM config used to extract GMM statistics on a per-segment basis
'gmm-regular = bob.bio.gmm.config.algorithm.gmm_regular:algorithm', 'gmm-regular = bob.bio.gmm.config.algorithm.gmm_regular:algorithm',
'jfa = bob.bio.gmm.config.algorithm.jfa:algorithm', 'jfa = bob.bio.gmm.config.algorithm.jfa:algorithm',
'isv = bob.bio.gmm.config.algorithm.isv:algorithm', 'isv = bob.bio.gmm.config.algorithm.isv:algorithm',
'ivector-cosine = bob.bio.gmm.config.algorithm.ivector_cosine:algorithm', 'ivector-cosine = bob.bio.gmm.config.algorithm.ivector_cosine:algorithm',
'ivector-plda = bob.bio.gmm.config.algorithm.ivector_plda:algorithm', 'ivector-plda = bob.bio.gmm.config.algorithm.ivector_plda:algorithm',
'ivector-lda-wccn-plda = bob.bio.gmm.config.algorithm.ivector_lda_wccn_plda:algorithm', 'ivector-lda-wccn-plda = bob.bio.gmm.config.algorithm.ivector_lda_wccn_plda:algorithm',
'ivector-segment = bob.bio.gmm.config.algorithm.ivector_segment:algorithm', # IVectors calculated on a per-segment basis
], ],
}, },
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment