print ('batch format: {0}'.format(batch_format)) print ('iterations upper limit: {0}'.format(iterations_upperlimit)) print ('validation function: {0}'.format(validation_func)) iterations_upperlimit = int(iterations_upperlimit) if batch_format == 'equal': equal_batch = True elif batch_format == 'random': equal_batch = False else: print ("Wrong batch_format option") exit() #------------------------------------ # Load Data # We should play a mask on image print ("starting time: {0}".format(time_stamp)) data, tracer, coords = astro_mnist.read_data_sets(images_name, labels_name, coords_name) print("Size of:") print("- Training-set:\t\t{}".format(len(data.train.labels))) print("- Test-set:\t\t{}".format(len(data.test.labels))) print("- Validation-set:\t{}".format(len(data.validation.labels))) #----------------------------------- # save arrangement and coords failure = save_arrangement(images_name[:-4], time_stamp, data, tracer) if not failure: print ("tracers and data are saved.") failure = save_coords(images_name[:-4], time_stamp, coords) if not failure: print ("coords are saved.") #----------------------------------- # Data dimension img_maj = imply_mask.count('0')
#-------------------------------------------- # main code if __name__ == "__main__": VERBOSE = 0 # measure times start_time = time.time() directory = argv[4] #----------------------------------- # Load Data images_name = argv[1] labels_name = argv[2] coords_name = argv[3] AI_saved_dir = argv[5] data, tracer, coords = astro_mnist.read_data_sets(images_name, labels_name, coords_name, train_weight=0, validation_weight=0, test_weight=1) print("Size of:") print("- Training-set:\t\t{}".format(len(data.train.labels))) print("- Test-set:\t\t{}".format(len(data.test.labels))) print("- Validation-set:\t{}".format(len(data.validation.labels))) data.test.cls = np.argmax(data.test.labels, axis=1) # save arrangement and coords failure = save_arrangement(images_name[:-4], directory, data, tracer) if not failure: print("tracer and data is saved.") failure = save_coords(images_name[:-4], directory, coords) if not failure: print("coords are saved.") #-----------------------------------