torchaudio 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 354a8636b2a04fd0ed60db93e2667f630f1db04ab047a6eb6e0c3607190c58d5
-  data.tar.gz: cde6229f8ba416c5a72bbea6d47e26fd987b6f13073f1ca9a1e6239cd62b79c1
+  metadata.gz: 2b395ce300252d4d65274f0ddb9d387de1dd836721a8a4f81804c5a888fb5433
+  data.tar.gz: 9e98ca9a0e8e945e069957990671bf143edb1240f313f65bf87e97bba4e05106
 SHA512:
-  metadata.gz: 76c9c637e4f12700c16bbf343104bb46f5e4f04161a3556cd0600ea236222902ac0ba3b9504f4bdb28a9919317a1b791c8f7b4dbcb48b35ed4c674de39a83277
-  data.tar.gz: '019d89cd2231c386c3458447358058618a6de78b77ed5b4d6d279ee0e3879946bda03cf339051136069dd552d6eeaed33bf40defa9d6c90c9a07eb27d68fdce2'
+  metadata.gz: ab922eef545e50e302f7a72f992f2d6040fb9cd1856c37ff532baed0092783442d649f8c3bc91a4cf92684729d7ce7294cd9a943a8c8a483734bb858e885845f
+  data.tar.gz: 414ea85a03b53cf21caef57289551fa88747ad2a5e0c60f0e92cb57a017cd04903b359d8d71ac37cecba50492678bc5d4327a9f340aba8f6006339d9379fed26

data/CHANGELOG.md

@@ -1,3 +1,9 @@
+## 0.1.2 (2021-02-06)
+
+- Added `amplitude_to_DB` and `DB_to_amplitude` methods
+- Added `AmplitudeToDB` transform
+- Fixed `save` options
+
 ## 0.1.1 (2020-08-26)
 
 - Added `save` method

data/LICENSE.txt

@@ -1,7 +1,7 @@
 BSD 2-Clause License
 
 Copyright (c) 2017 Facebook Inc. (Soumith Chintala),
-Copyright (c) 2020 Andrew Kane,
+Copyright (c) 2020-2021 Andrew Kane,
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without

data/README.md

@@ -2,7 +2,7 @@
 
 :fire: An audio library for Torch.rb
 
-[![Build Status](https://travis-ci.org/ankane/torchaudio.svg?branch=master)](https://travis-ci.org/ankane/torchaudio)
+[![Build Status](https://github.com/ankane/torchaudio/workflows/build/badge.svg?branch=master)](https://github.com/ankane/torchaudio/actions)
 
 ## Installation
 
@@ -56,6 +56,23 @@ Supported transforms are:
 - MuLawEncoding
 - Spectrogram
 
+## Functional
+
+```ruby
+TorchAudio::Functional.lowpass_biquad(waveform, sample_rate, cutoff_freq)
+```
+
+Supported functions are:
+
+- compute_deltas
+- dither
+- gain
+- highpass_biquad
+- lowpass_biquad
+- mu_law_decoding
+- mu_law_encoding
+- spectrogram
+
 ## Datasets
 
 Load a dataset

data/ext/torchaudio/ext.cpp

@@ -64,12 +64,12 @@ void Init_ext()
 
   Class rb_cSignalInfo = define_class_under<sox_signalinfo_t>(rb_mExt, "SignalInfo")
     .define_constructor(Constructor<sox_signalinfo_t>())
-    .define_method("rate", *[](sox_signalinfo_t self) { return self.rate; })
-    .define_method("channels", *[](sox_signalinfo_t self) { return self.channels; })
-    .define_method("precision", *[](sox_signalinfo_t self) { return self.precision; })
-    .define_method("length", *[](sox_signalinfo_t self) { return self.length; })
-    .define_method("rate=", *[](sox_signalinfo_t self, sox_rate_t rate) { self.rate = rate; })
-    .define_method("channels=", *[](sox_signalinfo_t self, unsigned channels) { self.channels = channels; })
-    .define_method("precision=", *[](sox_signalinfo_t self, unsigned precision) { self.precision = precision; })
-    .define_method("length=", *[](sox_signalinfo_t self, sox_uint64_t length) { self.length = length; });
+    .define_method("rate", *[](sox_signalinfo_t& self) { return self.rate; })
+    .define_method("channels", *[](sox_signalinfo_t& self) { return self.channels; })
+    .define_method("precision", *[](sox_signalinfo_t& self) { return self.precision; })
+    .define_method("length", *[](sox_signalinfo_t& self) { return self.length; })
+    .define_method("rate=", *[](sox_signalinfo_t& self, sox_rate_t rate) { self.rate = rate; })
+    .define_method("channels=", *[](sox_signalinfo_t& self, unsigned channels) { self.channels = channels; })
+    .define_method("precision=", *[](sox_signalinfo_t& self, unsigned precision) { self.precision = precision; })
+    .define_method("length=", *[](sox_signalinfo_t& self, sox_uint64_t length) { self.length = length; });
 }

data/lib/torchaudio.rb

@@ -20,6 +20,7 @@ require "torchaudio/transforms/mel_spectrogram"
 require "torchaudio/transforms/mu_law_encoding"
 require "torchaudio/transforms/mu_law_decoding"
 require "torchaudio/transforms/spectrogram"
+require "torchaudio/transforms/amplitude_to_db"
 require "torchaudio/version"
 
 module TorchAudio

data/lib/torchaudio/functional.rb

@@ -227,6 +227,19 @@ module TorchAudio
         output
       end
 
+      def amplitude_to_DB(amp, multiplier, amin, db_multiplier, top_db: nil)
+        db = Torch.log10(Torch.clamp(amp, min: amin)) * multiplier
+        db -= multiplier * db_multiplier
+
+        db = db.clamp(min: db.max.item - top_db) if top_db
+
+        db
+      end
+
+      def DB_to_amplitude(db, ref, power)
+        Torch.pow(Torch.pow(10.0, db * 0.1), power) * ref
+      end
+
       private
 
       def _apply_probability_distribution(waveform, density_function: "TPDF")

data/lib/torchaudio/transforms/amplitude_to_db.rb

@@ -0,0 +1,27 @@
+module TorchAudio
+  module Transforms
+    class AmplitudeToDB < Torch::NN::Module
+      def initialize(stype: :power, top_db: nil)
+        super()
+
+        @stype = stype
+
+        raise ArgumentError, 'top_db must be a positive numerical' if top_db && top_db.negative?
+
+        @top_db = top_db
+        @multiplier = stype == :power ? 10.0 : 20.0
+        @amin = 1e-10
+        @ref_value = 1.0
+        @db_multiplier = Math.log10([@amin, @ref_value].max)
+      end
+
+      def forward(amplitude_spectrogram)
+        F.amplitude_to_DB(
+          amplitude_spectrogram,
+          @multiplier, @amin, @db_multiplier,
+          top_db: @top_db
+        )
+      end
+    end
+  end
+end

data/lib/torchaudio/version.rb

@@ -1,3 +1,3 @@
 module TorchAudio
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torchaudio
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Andrew Kane
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-08-26 00:00:00.000000000 Z
+date: 2021-02-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: torch-rb
@@ -38,64 +38,8 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '2.2'
-- !ruby/object:Gem::Dependency
-  name: bundler
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: rake
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: rake-compiler
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '0'
-- !ruby/object:Gem::Dependency
-  name: minitest
-  requirement: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '5'
-  type: :development
-  prerelease: false
-  version_requirements: !ruby/object:Gem::Requirement
-    requirements:
-    - - ">="
-      - !ruby/object:Gem::Version
-        version: '5'
-description: 
-email: andrew@chartkick.com
+description:
+email: andrew@ankane.org
 executables: []
 extensions:
 - ext/torchaudio/extconf.rb
@@ -119,6 +63,7 @@ files:
 - lib/torchaudio/datasets/utils.rb
 - lib/torchaudio/datasets/yesno.rb
 - lib/torchaudio/functional.rb
+- lib/torchaudio/transforms/amplitude_to_db.rb
 - lib/torchaudio/transforms/mel_scale.rb
 - lib/torchaudio/transforms/mel_spectrogram.rb
 - lib/torchaudio/transforms/mu_law_decoding.rb
@@ -129,7 +74,7 @@ homepage: https://github.com/ankane/torchaudio
 licenses:
 - BSD-2-Clause
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -144,8 +89,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.2
-signing_key: 
+rubygems_version: 3.2.3
+signing_key:
 specification_version: 4
 summary: Data manipulation and transformation for audio signal processing
 test_files: []