1--batch=set_maskrcnn
2--batch=set_fastrcnn
3--batch=shapes_3d_unet
4