Updated readme to include model training and evaluation

README.md

@@ -58,11 +58,46 @@ $ cd robotic-grasping
 $ pip install -r requirements.txt
 ```
 
+## Datasets
+
+This repository supports both the [Cornell Grasping Dataset](http://pr.cs.cornell.edu/grasping/rect_data/data.php) and
+[Jacquard Dataset](https://jacquard.liris.cnrs.fr/).
+
+#### Cornell Grasping Dataset
+
+1. Download the and extract [Cornell Grasping Dataset](http://pr.cs.cornell.edu/grasping/rect_data/data.php). 
+2. Convert the PCD files to depth images by running `python -m utils.dataset_processing.generate_cornell_depth <Path To Dataset>`
+
+#### Jacquard Dataset
+
+1. Download and extract the [Jacquard Dataset](https://jacquard.liris.cnrs.fr/).
+
+
+## Model Training
+
+A model can be trained using the `train_network.py` script.  Run `train_network.py --help` to see a full list of options.
+
+For example:
+
+```bash
+python train_network.py --dataset cornell --dataset-path <Path To Dataset> --description training_cornell
+```
+
+## Model Evaluation
+
+The trained network can be evaluated using the `evaluate.py` script.  Run `evaluate.py --help` for a full set of options.
+
+For example:
+
+```bash
+python evaluate.py --network <Path to Trained Network> --dataset cornell --dataset-path <Path to Dataset> --iou-eval
+```
+
 ## Run Tasks
-Run the relevant task using the run programs. For example, to run the grasp generator run:
+A task can be executed using the relevant run script. All task scripts are named as `run_<task name>.py`. For example, to run the grasp generator run:
 ```bash
 python run_grasp_generator.py
 ```
 
 ## Run on a Robot
-Our ROS implementation for running the grasp generator with Baxter robot is available at: https://github.com/skumra/baxter-pnp
+To run the grasp generator with a robot, please use our ROS implementation for Baxter robot. It is available at: https://github.com/skumra/baxter-pnp