Tensorflow extracts image features - sample code