revised 2d unet, hand crafted feature