System and method for determining pupil center based on convolutional neural networks

A non-transitory computer readable storage medium storing instructions which, when executed by a processor, causes the processor to perform a method, the method comprising:

• obtaining a set of labeled pupil images, wherein a respective labeled pupil image comprises a pupil-segmentation label and a pupil-center-position label;
• constructing a multitask machine learning model that comprises a first branch for performing a pupil-region segmentation task and a second branch for performing a pupil-center-position regression task;
• and
• training the multitask machine learning model using the set of labeled pupil images;
• wherein training the multitask machine learning model comprises simultaneously training the first and second branches.

A computer system, comprising:

• a processor;
• and
• a storage device coupled to the processor, wherein the storage device storing instructions which, when executed by the processor, cause the processor to perform a method, the method comprising:
• obtaining a set of labeled pupil images, wherein a respective labeled pupil image comprises a pupil-segmentation label and a pupil-position label;
• constructing a multitask machine learning model that comprises a first branch for performing a pupil-region segmentation task and a second branch for performing a pupil-position regression task;
• and
• training the multitask machine learning model using the set of labeled pupil images;
• wherein training the multitask machine learning model comprises simultaneously training the first and second branches.