Summary: 3 instances, 3 unique

Text	Count
# FIXME: only augment train dataset, not the test dataset!	1
# TODO: Only augment training images, not entire dataset!	1
# TODO: make batch prediction work	1