Update README.md

capstone_traffic_light_classifier/README.md

@@ -1,15 +1,43 @@
 # Traffic Light Classifier
 
-Very simple traffic light classifier to be integrated in the capstone project.
+Simple traffic light classifier to be integrated in the capstone project.
 
 ---
 
-## Project Structure
+## Dataset
+
+### Simulator dataset
+**Number of frames:** 1324 frames
+
+**Resolution:** 800x600 pixel
+
+**Classes:** {none, red, yellow, green}
+
+**Samples:**
+![simulator_dataset](img/simulator_examples.png)
+
+
+### Udacity circuit dataset
+**Number of frames:** 1151 frames
 
-The traffic light dataset is modeled in [`traffic_light_dataset.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/traffic_light_dataset.py)
+**Resolution:** 800x600 pixel
 
-The deep network architecture is in [`traffic_light_classifier.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/traffic_light_classifier.py)
+**Classes:** {none, red, green}
 
-Training the model: [`train.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/train.py)
+**Samples:**
+![realworld_dataset](img/realworld_examples.png)
+
+### Mixed dataset
+
+The network has been from the dataset obtained merging the two dataset above.
+
+### Downloading the data
+
+All three datasets are available in this [shared folder](https://drive.google.com/open?id=0Bx9YaGcDPu3XSlMxeHlPZEIyRzQ).
+
+## Project Structure
 
-Testing the model from pretrained weights: [`test.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/test.py)
+- [`train.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/train.py): entry point for training the model
+- [`test.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/test.py): entry point for testing the model from pretrained weights
+- [`traffic_light_dataset.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/traffic_light_dataset.py): contains`TrafficLightDataset` class to abstract over the raw data
+- [`traffic_light_classifier.py`](https://github.com/ndrplz/self-driving-car/blob/master/capstone_traffic_light_classifier/traffic_light_classifier.py): contains `TrafficLightClassifier` deep network for classification