Make more general to support all net training

anson0910 · Nov 28, 2015 · dfef95a · dfef95a
1 parent 6493c1b
commit dfef95a
Showing 1 changed file with 29 additions and 24 deletions.
diff --git a/face_preprocess_10kUS/write_train_val.py b/face_preprocess_10kUS/write_train_val.py
@@ -3,36 +3,41 @@
 import shutil
 import random
 
+trainingNet = 12
+
 # load and open files to read and write
 # =================== face_12c =================================
-# pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
-# neg_file_name = '/home/anson/caffe-master/data/face/all_negatives.txt'
-# train_file_name = '/home/anson/face_pictures/train_12c'
-# val_file_name = '/home/anson/face_pictures/val'
-# write_train_name = '/home/anson/caffe-master/data/face/train_12c.txt'
-# write_train = open(write_train_name, "w")
-# write_val_name = '/home/anson/caffe-master/data/face/val.txt'
-# write_val = open(write_val_name, "w")
+if trainingNet == 12:
+ pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
+ neg_file_name = '/home/anson/caffe-master/data/face/all_negatives.txt'
+ train_file_name = '/home/anson/face_pictures/train_12c'
+ val_file_name = '/home/anson/face_pictures/val'
+ write_train_name = '/home/anson/caffe-master/data/face/train_12c.txt'
+ write_train = open(write_train_name, "w")
+ write_val_name = '/home/anson/caffe-master/data/face/val.txt'
+ write_val = open(write_val_name, "w")
 
 # =================== face_24c =================================
-# pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
-# neg_file_name = '/home/anson/caffe-master/data/face/all_negatives_24c.txt'
-# train_file_name = '/home/anson/face_pictures/train_24c'
-# val_file_name = '/home/anson/face_pictures/val'
-# write_train_name = '/home/anson/caffe-master/data/face/train_24c.txt'
-# write_train = open(write_train_name, "w")
-# write_val_name = '/home/anson/caffe-master/data/face/val.txt'
-# write_val = open(write_val_name, "w")
+elif trainingNet == 24:
+ pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
+ neg_file_name = '/home/anson/caffe-master/data/face/all_negatives_24c.txt'
+ train_file_name = '/home/anson/face_pictures/train_24c'
+ val_file_name = '/home/anson/face_pictures/val'
+ write_train_name = '/home/anson/caffe-master/data/face/train_24c.txt'
+ write_train = open(write_train_name, "w")
+ write_val_name = '/home/anson/caffe-master/data/face/val.txt'
+ write_val = open(write_val_name, "w")
 
 # =================== face_48c =================================
-pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
-neg_file_name = '/home/anson/caffe-master/data/face/all_negatives_48c.txt'
-train_file_name = '/home/anson/face_pictures/train_48c'
-val_file_name = '/home/anson/face_pictures/val'
-write_train_name = '/home/anson/caffe-master/data/face/train_48c.txt'
-write_train = open(write_train_name, "w")
-write_val_name = '/home/anson/caffe-master/data/face/val.txt'
-write_val = open(write_val_name, "w")
+elif trainingNet == 48:
+ pos_file_name = '/home/anson/caffe-master/data/face/all_positives.txt'
+ neg_file_name = '/home/anson/caffe-master/data/face/all_negatives_48c.txt'
+ train_file_name = '/home/anson/face_pictures/train_48c'
+ val_file_name = '/home/anson/face_pictures/val'
+ write_train_name = '/home/anson/caffe-master/data/face/train_48c.txt'
+ write_train = open(write_train_name, "w")
+ write_val_name = '/home/anson/caffe-master/data/face/val.txt'
+ write_val = open(write_val_name, "w")
 
 
 pos = []