Narrow down the datas to be trained

The graph cut is run 20*15*15 times, if the data is too large, the
training will become extremely time consuming.