gtcrn 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 684bab48a155c6ad43d42479fd8c361632a4c51d1ecf19f388e7487e575c3ce4
-  data.tar.gz: 78655c56cf93b6839ba193112534c02a4e8d73084c89ec6536dc9db6cfb12032
+  metadata.gz: e62dd7f2ba7da50ab841769e0ccc9d0aeef8f0a31499a2a59b80ffc749b1ca9b
+  data.tar.gz: 555a5c91d412822e1e066bb1ea4bccdfd1cdf9e334adad025b886aea3a2b62f3
 SHA512:
-  metadata.gz: e71548aba12e685361ab24a6f2a1574a151cce46c622bf281f54254273b9f8e83ad7ced377ff0ad0b9096e530fef2c1dab289f6a3f1bd3457d511599960ecc64
-  data.tar.gz: a8ad7d5e076c0257a79a7952fc96ab15f001280113cbc05f7e0d49d6b4ddb95ae562a02ffc15e25d94b4b8b84a6390eaaae434cd5899b91fd0622af1a2e1f43f
+  metadata.gz: c82eb21211da591054d65de5aa131219432f20ce161b53b750d961baed6bd5e3f840076ff17dcdf9568a059257544de648a33c1f7d51911e0513c2e855bb3e2e
+  data.tar.gz: 50814826a17fb42e874f7ea7bd8fae2366f4b08d2961c2e354f4b8b9e4346199d37e61495c4a236cafc46f9886c2a1a9ba6f4b4e9b1b5f5e6f837a0a16265f03

data/README.md

@@ -16,14 +16,14 @@ output = GTCRN.new.enhance_speech("path/to/audio.wav")
 # => <Pathname:path/to/audio.enhanced.wav>
 ````
 
-Audio file must be monoral WAV with 16kHz sampling rate and 16-bit per sample..
+Audio file must be mono WAV with 16kHz sampling rate and 16-bit per sample.
 
 CLI
 ---
 
 This gem ships with `gtcrn` command.
 
-    % gtcrn path/to/audio.wav path/to/output.wav
+    % gtcrn path/to/audio.wav --output=path/to/output.wav
     Enhanced file written to
     path/to/output.wav
 

data/bin/gtcrn

@@ -1,11 +1,28 @@
+require "optparse"
 require "gtcrn"
 
 def main(argv)
+  options = parse_options(argv)
   input = argv.shift
   abort "Specify input file" unless input
-  output = GTCRN.new.enhance_speech(input, argv.shift)
+  output = GTCRN.new.enhance_speech(input, options[:output])
   $stderr.puts "Enhanced file written to"
   $stderr.puts output
 end
 
+def parse_options(argv)
+  options = {}
+  OptionParser.new {|opt|
+    opt.banner = <<~EOB
+      Enhance speech properties of GTCRN files
+
+      Usage: gtcrn [options] INPUT
+    EOB
+    opt.on "-o", "--output=PATH", "Specify output file or directory" do |path|
+      options[:output] = path
+    end
+  }.parse!(argv)
+  options
+end
+
 main ARGV

data/gtcrn.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = "gtcrn"
-  s.version = "0.0.1"
+  s.version = "0.0.2"
   s.authors = ["Kitaiti Makoto"]
   s.email = ["KitaitiMakoto@gmail.com"]
   s.summary = "Denoises audio"

data/lib/gtcrn.rb

@@ -35,7 +35,7 @@ class GTCRN
     waveform, sample_rate = TorchAudio.load(path.to_path)
     raise "Sampling rate must be 16000 Hz, but given: #{sample_rate} Hz" unless sample_rate == 16000
     enhanced = enhance_speech_waveform(waveform)
-    TorchAudio.save(dest.to_path, enhanced.squeeze, sample_rate)
+    TorchAudio.save(dest.to_path, enhanced, sample_rate)
 
     dest
   end
@@ -66,6 +66,6 @@ class GTCRN
       Torch.from_numo(real) + 1i * Torch.from_numo(imag),
       **ISTFT_OPTS
     )
-    enhanced
+    enhanced.squeeze
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gtcrn
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Kitaiti Makoto