diarize-ruby 0.3.0

data/test/segment_test.rb

@@ -0,0 +1,29 @@
+require 'test_helper'
+require 'ostruct'
+require 'uri'
+
+class SegmentTest < Test::Unit::TestCase
+
+  def test_initialize
+    segment = Diarize::Segment.new('audio', 'start', 'duration', 'gender', 'bandwidth', 'speaker_id')
+    assert_equal segment.instance_variable_get('@audio'), 'audio'
+    assert_equal segment.instance_variable_get('@start'), 'start'
+    assert_equal segment.instance_variable_get('@duration'), 'duration'
+    assert_equal segment.instance_variable_get('@speaker_gender'), 'gender'
+    assert_equal segment.instance_variable_get('@bandwidth'), 'bandwidth'
+    assert_equal segment.instance_variable_get('@speaker_id'), 'speaker_id'
+  end
+
+  def test_speaker
+    segment = Diarize::Segment.new(OpenStruct.new({:base_uri => 'http://example.com'}), nil, nil, 'm', nil, 's1') 
+    assert_equal segment.speaker.object_id, segment.speaker.object_id # same one should be generated twice
+    assert_equal segment.speaker.uri, URI('http://example.com#s1')
+    assert_equal segment.speaker.gender, 'm'
+  end
+
+  def test_uri
+    segment = Diarize::Segment.new(OpenStruct.new({:base_uri => 'http://example.com'}), 2, 5, 'm', nil, 's1')
+    assert_equal segment.uri, URI('http://example.com#t=2,7')
+  end
+
+end

data/test/segmentation_test.rb

@@ -0,0 +1,39 @@
+require 'test_helper'
+require 'tempfile'
+
+class SegmentationTest < Test::Unit::TestCase
+
+  def test_segmentation_from_empty_seg_file
+    audio_uri = URI('file:' + File.join(File.dirname(__FILE__), 'data', 'foo.wav'))
+    audio = Diarize::Audio.new audio_uri
+    seg_file = Tempfile.new(['diarize-jruby', '.seg'])
+    seg_file.write ''
+    seg_file.close
+    segmentation = Diarize::Segmentation.from_seg_file(audio, seg_file.path)
+    assert_equal segmentation.size, 0
+  end
+
+  def test_segmentation_from_sef_file
+    audio_uri = URI('file:' + File.join(File.dirname(__FILE__), 'data', 'foo.wav'))
+    audio = Diarize::Audio.new audio_uri
+    seg_file = Tempfile.new(['diarize-jruby', '.seg'])
+    seg_file.write <<EOF
+foo 1 0 1000 F S U S0
+foo 1 1000 10000 M S U S1
+EOF
+    seg_file.close
+    segmentation = Diarize::Segmentation.from_seg_file(audio, seg_file.path)
+    assert_equal segmentation.size, 2
+    assert_equal segmentation.first.class, Diarize::Segment
+    assert_equal segmentation.first.start, 0
+    assert_equal segmentation.first.duration, 10
+    assert_equal segmentation.first.speaker.uri, URI(audio_uri.to_s + '#S0')
+    assert_equal segmentation.first.speaker.gender, 'F'
+    assert_equal segmentation.last.class, Diarize::Segment
+    assert_equal segmentation.last.start, 10
+    assert_equal segmentation.last.duration, 100
+    assert_equal segmentation.last.speaker.uri, URI(audio_uri.to_s + '#S1')
+    assert_equal segmentation.last.speaker.gender, 'M'
+  end
+
+end

data/test/speaker_test.rb

@@ -0,0 +1,101 @@
+require 'test_helper'
+require 'tempfile'
+
+class SpeakerTest < Test::Unit::TestCase
+
+  def test_detection_threshold
+    Diarize::Speaker.detection_threshold = 0.1
+    assert_equal 0.1, Diarize::Speaker.detection_threshold
+  end
+
+  def test_find_or_create_gives_same_object_if_called_with_same_id
+    speaker1 = Diarize::Speaker.find_or_create('S0', 'M')
+    speaker2 = Diarize::Speaker.find_or_create('S0', 'M')
+    assert_equal speaker2.object_id, speaker1.object_id
+  end
+
+  def test_initialize
+    speaker = Diarize::Speaker.new('uri', 'm')
+    assert_equal speaker.uri, 'uri'
+    assert_equal speaker.gender, 'm'
+  end
+
+  def test_initialize_ubm
+    speaker = Diarize::Speaker.ubm
+    assert_equal speaker.gender, nil
+    assert_equal speaker.uri, nil
+    assert_equal speaker.model.name, 'MSMTFSFT' # UBM GMM
+  end
+
+  def test_initialize_with_model
+    model_file = File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm')
+    speaker = Diarize::Speaker.new(nil, nil, model_file)
+    assert_equal speaker.model.name, 'S0'
+  end
+
+  def test_mean_log_likelihood
+    speaker = Diarize::Speaker.ubm
+    assert speaker.mean_log_likelihood.nan?
+    speaker.mean_log_likelihood = 1
+    assert_equal speaker.mean_log_likelihood, 1
+  end
+
+  def test_supervector
+    speaker = Diarize::Speaker.new(nil, nil, File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm'))
+    assert_equal 512 * 24, speaker.supervector.dim
+    # Testing the first and the last elements are OK
+    assert_equal speaker.model.components.get(0).mean(0), speaker.supervector.vector[0]
+    assert_equal speaker.model.components.get(511).mean(23), speaker.supervector.vector[512 * 24 - 1]
+  end
+
+  def test_save_and_load_model
+    speaker = Diarize::Speaker.ubm
+    tmp = Tempfile.new(['diarize-test', '.gmm'])
+    speaker.save_model(tmp.path)
+    model = Diarize::Speaker.load_model(tmp.path)
+    assert_equal speaker.model.components.get(0).mean(0), model.components.get(0).mean(0)
+    File.delete(tmp.path)
+  end
+
+  def test_divergence_returns_nil_if_one_model_is_empty
+    speaker1 = Diarize::Speaker.ubm
+    speaker2 = Diarize::Speaker.ubm
+    speaker2.model = nil
+    assert_equal Diarize::Speaker.divergence(speaker1, speaker2), nil
+    assert_equal Diarize::Speaker.divergence(speaker2, speaker1), nil
+  end
+
+  def test_divergence_is_symmetric
+    model_file = File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm')
+    speaker1 = Diarize::Speaker.new(nil, nil, model_file)
+    speaker2 = Diarize::Speaker.ubm
+    assert Diarize::Speaker.divergence(speaker1, speaker2) > 0
+    assert_equal Diarize::Speaker.divergence(speaker1, speaker2), Diarize::Speaker.divergence(speaker2, speaker1)
+    assert_equal Diarize::Speaker.divergence(speaker1, speaker1), 0.0
+  end
+
+  def test_divergence_ruby_is_same_as_divergence_lium
+    model_file = File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm')
+    speaker1 = Diarize::Speaker.new(nil, nil, model_file)
+    speaker2 = Diarize::Speaker.ubm
+    assert_equal Diarize::Speaker.divergence_lium(speaker1, speaker2).round(12), Diarize::Speaker.divergence_ruby(speaker1, speaker2).round(12)
+  end
+
+  def test_normalise
+    # Testing M-Norm
+    model_file = File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm')
+    speaker1 = Diarize::Speaker.new(nil, nil, model_file)
+    speaker2 = Diarize::Speaker.ubm
+    assert Diarize::Speaker.divergence(speaker1, speaker2) != 1.0
+    speaker1.normalize! # Putting speaker1.gmm at distance 1 from UBM
+    assert Diarize::Speaker.divergence(speaker1, speaker2) - 1.0 < 1e-12 # rounding error
+  end
+
+  def test_do_not_normalize_ubm
+    speaker = Diarize::Speaker.ubm
+    old_supervector = speaker.supervector
+    speaker.normalize!
+    assert_equal old_supervector, speaker.supervector
+  end
+
+end

data/test/super_vector_test.rb

@@ -0,0 +1,24 @@
+require 'test_helper'
+
+class SuperVectorTest < Test::Unit::TestCase
+
+  def test_generate_from_model
+    model = Diarize::Speaker.load_model(File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm'))
+    sv = Diarize::SuperVector.generate_from_model(model)
+    assert_equal 512 * 24, sv.dim
+    # Checking all elements are OK
+    model.nb_of_components.times do |i|
+      gaussian = model.components.get(i)
+      gaussian.dim.times do |j|
+        assert_equal gaussian.mean(j), sv.vector[i * gaussian.dim + j]
+      end
+    end
+  end
+
+  def test_hash
+    model = Diarize::Speaker.load_model(File.join(File.dirname(__FILE__), 'data', 'speaker1.gmm'))
+    sv = Diarize::SuperVector.generate_from_model(model)
+    assert_equal sv.vector.hash, sv.hash
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,23 @@
+require 'rubygems'
+require 'test/unit'
+require 'test/unit/assertions'
+require 'diarize'
+require 'mocha/test_unit'
+require 'webmock/test_unit'
+
+WebMock.disable_net_connect!(:net_http_connect_on_start => true)
+
+class Test::Unit::TestCase
+
+  # Add global extensions to the test case class here
+
+  # E.g. "/Users/foo/work/test"
+  def test_root
+    File.dirname(__FILE__)
+  end
+
+  def fixtures_root
+    "#{test_root}/data"
+  end
+
+end

metadata

@@ -0,0 +1,168 @@
+--- !ruby/object:Gem::Specification
+name: diarize-ruby
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+platform: ruby
+authors:
+- Yves Raimond
+- Juergen Fesslmeier
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-07-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: test-unit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: rjb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.5'
+- !ruby/object:Gem::Dependency
+  name: to-rdf
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jblas-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.1'
+description: A library for Ruby wrapping the LIUM Speaker Diarization and including
+  a few extra tools
+email:
+- jfesslmeier@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".ruby-gemset"
+- ".ruby-version"
+- AUTHORS
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- diarize-ruby.gemspec
+- lib/diarize.rb
+- lib/diarize/LIUM_SpkDiarization-4.2.jar
+- lib/diarize/audio.rb
+- lib/diarize/audio_player.rb
+- lib/diarize/lium.rb
+- lib/diarize/segment.rb
+- lib/diarize/segmentation.rb
+- lib/diarize/speaker.rb
+- lib/diarize/super_vector.rb
+- lib/diarize/ubm.gmm
+- lib/diarize/version.rb
+- test/audio_test.rb
+- test/data/foo.wav
+- test/data/speaker1.gmm
+- test/data/will-and-juergen.wav
+- test/segment_test.rb
+- test/segmentation_test.rb
+- test/speaker_test.rb
+- test/super_vector_test.rb
+- test/test_helper.rb
+homepage: https://github.com/chinshr/diarize-ruby
+licenses:
+- GNU Affero General Public License version 3
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.8
+signing_key: 
+specification_version: 4
+summary: Speaker Diarization for Ruby
+test_files:
+- test/audio_test.rb
+- test/data/foo.wav
+- test/data/speaker1.gmm
+- test/data/will-and-juergen.wav
+- test/segment_test.rb
+- test/segmentation_test.rb
+- test/speaker_test.rb
+- test/super_vector_test.rb
+- test/test_helper.rb