Closed waveroarer closed 6 years ago
Hello, I got an error,
Code:
import csv import cv2 import numpy as np import sklearn import os from random import shuffle lines = [] with open('./data/driving_log.csv') as csvfile: reader = csv.reader(csvfile) for line in reader: lines.append(line) images = [] measurements = [] correction = 0.15 from sklearn.model_selection import train_test_split train_samples, validation_samples = train_test_split(lines, test_size=0.2) def generator(samples, batch_size=32): num_samples = len(lines) while 1: # Loop forever so the generator never terminates shuffle(samples) for offset in range(0, num_samples, batch_size): batch_samples = samples[offset:offset+batch_size] images = [] angles = [] for batch_sample in batch_samples: name = './IMG/'+batch_sample[0].split('\\')[-1] center_image = cv2.imread(name) center_angle = float(batch_sample[3]) images.append(center_image) angles.append(center_angle) # trim image to only see section with road X_train = np.array(images) y_train = np.array(angles) yield sklearn.utils.shuffle(X_train, y_train) from keras.models import Sequential from keras.layers import Cropping2D from keras.layers import Flatten, Dense, Lambda from keras.layers.convolutional import Convolution2D from keras.layers.pooling import MaxPooling2D train_generator = generator(train_samples, batch_size=32) validation_generator = generator(validation_samples, batch_size=32) model = Sequential() model.add(Lambda(lambda x: x / 127.5 - 1, input_shape=(160,320,3),output_shape=(160,320,3))) model.add(Cropping2D(cropping=((70,25),(0,0)))) model.add(Convolution2D(24,5,5,subsample=(2,2),activation="relu")) model.add(Convolution2D(36,5,5,subsample=(2,2),activation="relu")) model.add(Convolution2D(48,5,5,subsample=(2,2),activation="relu")) model.add(Convolution2D(64,3,3,activation="relu")) model.add(Convolution2D(64,3,3,activation="relu")) model.add(Flatten()) model.add(Dense(100)) model.add(Dense(50)) model.add(Dense(10)) model.add(Dense(1)) model.compile(loss='mse',optimizer='adam') model.fit_generator(train_generator, samples_per_epoch= len(train_samples), validation_data=validation_generator, nb_val_samples=len(validation_samples), nb_epoch=3) model.save('model.h5')
Error:
ValueError Traceback (most recent call last)
Thank you for letting us know. It isn't clear that this is a bug. Would you mind discussing with a mentor? If it is a bug, let us know, and we can re-open the issue.
Hello, I got an error,
Code:
Error:
_Epoch 1/3
ValueError Traceback (most recent call last)