changed test image to kg, added additional output sizes