some improvement:
-  let batchsize=2, can speed up the testing process, for the dataset is only contain 2 images.
- polish the final result: draw a red mask on orginal image.