Make more general to support all net training

face_preprocess_10kUS/shuffle_write_negatives.py

@@ -1,15 +1,31 @@
 import os
 import random
 
+trainingNet = 12
+
 data_base_dir = "/home/anson/face_pictures/negatives" # directory containing files of positives
-start_neg_dir = 98
-end_neg_dir = 98
-file_list = [] # list to save image names
 
-# load and open files to read and write
-write_file_name = '/home/anson/caffe-master/data/face/all_negatives_48c.txt'
+if trainingNet == 12:
+ start_neg_dir = 1
+ end_neg_dir = 33
+ # load and open files to read and write
+ write_file_name = '/home/anson/caffe-master/data/face/all_negatives.txt'
+elif trainingNet == 24:
+ start_neg_dir = 98
+ end_neg_dir = 98
+ # load and open files to read and write
+ write_file_name = '/home/anson/caffe-master/data/face/all_negatives_24c.txt'
+elif trainingNet == 48:
+ start_neg_dir = 99
+ end_neg_dir = 99
+ # load and open files to read and write
+ write_file_name = '/home/anson/caffe-master/data/face/all_negatives_48c.txt'
+
+
 write_file = open(write_file_name, "w")
 
+file_list = [] # list to save image names
+
 for current_neg_dir in range(start_neg_dir, end_neg_dir + 1):
  current_dir = data_base_dir + '/negative_' + str(current_neg_dir).zfill(2)