How many images are there in one training epoch?

[ygtxr1997]

Your paper says that 'Mix-WebFace is the mixup of Occ-WebFace and WebFace with the ratio 2:1'.
How many images in the Occ-WebFace?
How many images in one training epoch after mixing up Occ-WebFace and WebFace?

[haibo-qiu]

Hi @ygtxr1997

Our statement 'Mix-WebFace is the mixup of Occ-WebFace and WebFace with the ratio 2:1' might be a little bit confusing. The practical situation is that when iterating images in WebFace, each image has a 2/3 probability of being masked and a 1/3 probability of keeping clean. You can refer to the below code where the self.ratio is 3.

FROM/lib/datasets/dataset.py

Lines 175 to 180 in 6b825c6

	elif self.mode == 'Mask' or self.mode == 'Occ':
	if random.choice(range(self.ratio)) == 0:
	return img, 0
	else:
	img_occ, mask_label, mask = self.occlude_img(img)
	return img_occ, mask_label

Thus the total images in one epoch are the same as WebFace images, nearly 500k.

The Occ-WebFace means masking each image in WebFace with the random scale from 1:0.5:5 (1, 1.5, 2.0....5.0) when training, which also leads to the same account of images (500k).

[ygtxr1997]

Thank you very much!