download=True option

stanford_mir/core.py

@@ -64,7 +64,7 @@ def get_features(collection='drum_samples_train',
 
         return test_features, labels, scaler
 
-def download_samples(collection='drum_samples_train'):
+def download_samples(collection='drum_samples_train', download=True):
     """Download ten kick drum samples and ten snare drum samples.
 
         `collection`: output directory containing the twenty drum samples
@@ -80,9 +80,9 @@ def download_samples(collection='drum_samples_train'):
     except OSError as exception:
         if exception.errno != errno.EEXIST:
             raise
-        else:
-            print "Directory %s already exists." % collection
+
     if collection == 'drum_samples_train':
+        if download:
             for drum_type in ['kick', 'snare']:
                 for i in range(1, 11):
                     filename = '%s_%02d.wav' % (drum_type, i)
@@ -91,7 +91,9 @@ def download_samples(collection='drum_samples_train'):
         kick_filepaths = [os.path.join(collection, 'kick_%02d.wav' % i) for i in range(1, 11)]
         snare_filepaths = [os.path.join(collection, 'snare_%02d.wav' % i) for i in range(1, 11)]
         return kick_filepaths, snare_filepaths
+
     elif collection == 'drum_samples_test':
+        if download:
             for drum_type in ['kick', 'snare']:
                 for i in range(30):
                     filename = '%s_%02d.wav' % (drum_type, i)
@@ -100,3 +102,14 @@ def download_samples(collection='drum_samples_train'):
         kick_filepaths = [os.path.join(collection, 'kick_%02d.wav' % i) for i in range(30)]
         snare_filepaths = [os.path.join(collection, 'snare_%02d.wav' % i) for i in range(30)]
         return kick_filepaths, snare_filepaths
+
+    elif collection == 'violin_samples_train':
+        urllib.urlretrieve('http://audio.musicinformationretrieval.com/violin_samples_train/list.txt',
+                           filename=os.path.join(collection, 'list.txt'))
+        for line in open(os.path.join(collection, 'list.txt'), 'r'):
+            filename = line.strip()
+            print filename
+            if filename.endswith('.wav'):
+                urllib.urlretrieve('http://audio.musicinformationretrieval.com/' + filename,
+                                   filename=filename)
+        return [os.path.join(collection, f) for f in os.listdir(collection)]