a96bf92790895e2f1a83ce9f9fbc7cbca38fb5b3,examples/transferlearning_model.py,,,#,2

Before Change

csv_file_path = "image_data/sample.csv"
imageProcessor = ImageProcessor(from_csv=True, target_labels=target_labels, datapath=csv_file_path, target_dimensions=target_dimensions, raw_dimensions=raw_dimensions, csv_label_col=0, csv_image_col=1)

features, labels = imageProcessor.get_training_data()

print("--------------- Inception-V3 Model -------------------")
print("Initializing neural network with InceptionV3 base model...")
model = TransferLearningNN(model_name=model_name, target_labels=target_labels)

After Change

    print("raw image shape: " + str(images.shape))

print("Processing data...")
imageProcessor = ImageProcessor(images, target_dimensions=target_dimensions)
images = imageProcessor.process_training_data()
if verbose:
    print ("processed image shape: " + str(images.shape))

print("--------------- Inception-V3 Model -------------------")
print("Initializing neural network with InceptionV3 base model...")
model = TransferLearningNN(model_name=model_name, target_labels=target_labels)

Instances