do not trim off recording file extension when naming generated files

Author: bjarthur

app/post-process.py

@@ -1,3 +1,5 @@
+#!/usr/bin/env python3
+
 import sys
 import ast
 import os
@@ -25,9 +27,7 @@
 
 M.init(None, os.path.join(__dir__, "configuration.py"), False)
 
-wavfile_noext = M.trim_ext(wavfile)
-
-with open(wavfile_noext+"-classify.log",'r') as fid:
+with open(wavfile+"-classify.log",'r') as fid:
     for line in fid:
         if "labels: " in line:
             m=re.search('labels: (.+)',line)
@@ -40,13 +40,13 @@
 
 if delete_wavs:
     for label in labels:
-        fullpath = wavfile_noext+'-'+label+'.wav'
+        fullpath = wavfile+'-'+label+'.wav'
         os.remove(fullpath)
         print("deleting ", fullpath)
 
 durations = {}
 counts = {}
-with open(wavfile_noext+'-predicted-1.0pr.csv') as fid:
+with open(wavfile+'-predicted-1.0pr.csv') as fid:
     csvreader = csv.reader(fid)
     for row in csvreader:
         if row[4] not in durations:
@@ -55,7 +55,7 @@
         durations[row[4]] += (int(row[2]) - int(row[1]))
         counts[row[4]] += 1
 
-with open(wavfile_noext+"-post-process.csv",'w') as fid:
+with open(wavfile+"-post-process.csv",'w') as fid:
     fid.write("wavfile,label,duration ("+M.context_time_units+"),num events\n")
     for label in durations.keys():
         fid.write(os.path.basename(wavfile)+','+
@@ -64,8 +64,8 @@
                   str(counts[label])+'\n')
 
 if delete_logs:
-    os.remove(wavfile_noext+'-classify.log')
-    os.remove(wavfile_noext+'-ethogram.log')
-    os.remove(wavfile_noext+'-post-process.log')
+    os.remove(wavfile+'-classify.log')
+    os.remove(wavfile+'-ethogram.log')
+    os.remove(wavfile+'-post-process.log')
 
 print(str(datetime.now())+": finish time")

…0161207T102314_ch1-annotated-notsong.csv …207T102314_ch1.wav-annotated-notsong.csvdata/20161207T102314_ch1-annotated-notsong.csv renamed to data/20161207T102314_ch1.wav-annotated-notsong.csv data/20161207T102314_ch1-annotated-notsong.csv renamed to data/20161207T102314_ch1.wav-annotated-notsong.csv

@@ -84,7 +84,7 @@ def main():
 
   load_audio_read_plugin(FLAGS.audio_read_plugin, FLAGS.audio_read_plugin_kwargs)
   load_video_read_plugin(FLAGS.video_read_plugin, FLAGS.video_read_plugin_kwargs)
-  from lib import audio_read, video_read, trim_ext
+  from lib import audio_read, video_read
 
   with open(FLAGS.model_labels, 'r') as fid:
     model_labels = fid.read().splitlines()
@@ -255,13 +255,11 @@ def main():
         else:
           adjusted_probability = probability_matrix[:,ch]
         waveform = adjusted_probability*np.iinfo(np.int16).max
-        withoutext = trim_ext(FLAGS.wav)
-        filename = withoutext+'-'+labels[ch]+'.wav'
+        filename = FLAGS.wav+'-'+labels[ch]+'.wav'
         wavfile.write(filename, int(sample_rate), waveform.astype('int16'))
   else:
         waveform = probability_matrix * np.iinfo(np.int16).max
-        withoutext = trim_ext(FLAGS.wav)
-        filename = withoutext+'-.wav'
+        filename = FLAGS.wav+'-.wav'
         wavfile.write(filename, int(sample_rate), waveform.astype('int16'))
 
 if __name__ == '__main__':