python3 retrain.py --image_dir dataset --output_graph output_graph.pb --output_labels output_labels.txt
