created data for training and testing
still testing lemon detection