Update sonogram_utility.py

sonogram_utility.py

@@ -119,10 +119,13 @@ def splitIntoTimeSegments(testFile,maxDurationInSeconds=60):
     return audioSegments, sample_rate
 
 def audioNormalize(waveform,sampleRate,stepSizeInSeconds = 2,dbThreshold = -50,dbTarget = -5):
+    print("In audioNormalize")
     copyWaveform = waveform.clone().detach()
     copyWaveform_db = waveform.clone().detach()
+    print("Copies made")
     transform = torchaudio.transforms.AmplitudeToDB(stype="amplitude", top_db=80)
     copyWaveform_db = transform(copyWaveform_db)
+    print("DB levels calculated")
     currStart = 0
     currEnd = int(min(currStart + stepSizeInSeconds * sampleRate, len(copyWaveform_db[0])-1))
     done = False
@@ -141,10 +144,12 @@ def audioNormalize(waveform,sampleRate,stepSizeInSeconds = 2,dbThreshold = -50,d
             done = True
         else:
             currEnd = int(min(currStart + stepSizeInSeconds * sampleRate, len(copyWaveform_db[0])-1))
+    print("Waveform enhanced")
     return copyWaveform
 
 class equalizeVolume(torch.nn.Module):
     def forward(self, waveform,sampleRate,stepSizeInSeconds,dbThreshold,dbTarget):
+        print("In equalizeVolume")
         waveformDifference = audioNormalize(waveform,sampleRate,stepSizeInSeconds,dbThreshold,dbTarget)
         return waveformDifference