Computer vision and defect detection in TensorFlow