correction get_spectrogram

get_spectrogram.py

@@ -43,15 +43,17 @@ def main(data, arguments):
             # Check if empty .txt annotation file is needed
             if arguments.background:
                 folder = 'background'
+                utils.create_directory(os.path.join(arguments.directory, folder))
                 name = os.path.join(arguments.directory, folder, f'{file}_{offset}')
                 empty_dataframe = pd.DataFrame(columns=['specie', 'x', 'y', 'w', 'h'])
                 empty_dataframe.to_csv(str(name+'.txt'), sep=' ', index=False, header=False)
             else:
                 folder = 'spectrograms'
+                utils.create_directory(os.path.join(arguments.directory, folder))
                 name = os.path.join(arguments.directory, folder, f'{file}_{offset}')
             
             utils.create_spectrogram(
-                sig, arguments.directory, name, window_size=arguments.window,
+                sig, arguments.directory, name, arguments.cmap, window_size=int(arguments.window),
                 overlap=arguments.hop)
             
         except Exception as error:
@@ -73,6 +75,8 @@ if __name__ == "__main__":
                         default=1024)
     parser.add_argument('--hop', type=float,
                         help='Ratio of hop in window : 50%% = 0.5', default=0.5)
+    parser.add_argument('--cmap', type=str, help='Colormar of the spectrograms', 
+                        choices=['jet', 'cividis', 'viridis'], default='viridis')
     parser.add_argument('--high', type=int,
                         help='High Pass Filter value in Hz', default=10)
     parser.add_argument('--low', type=int,