Embodiments of the present disclosure include a method, device and computer readable medium involving receiving image data of one or more coronary arteries, generating a binary segmentation indicating presence of calcium in the one or more coronary arteries from the image data, generating a branch density of the one or more coronary arteries, and assigning a coronary artery label from the branch density to the binary segmentation such that at least one indication of presence of calcium of the binary segmentation is labeled as present in a specific one of the one or more coronary arteries.