Single shot model, COVID-19 prediction + lesion detection/segmentation