gluonfr.data

Hint

Please refer to Datasets for the description of the datasets listed in this page, and how to download and extract them.

API Reference

This module provides popular face recognition datasets.

class gluonfr.data.FRTrainRecordDataset

A dataset wrapping over a rec serialized file provided by InsightFace Repo.

Parameters:

• name (str. Name of val dataset.) –
• root (str. Path to face folder. Default is '$(HOME)/mxnet/datasets/face') –
• transform (function, default None) –

  A user defined callback that transforms each sample. For example:

  transform=lambda data, label: (data.astype(np.float32)/255, label)
  

class gluonfr.data.FRValDataset

A dataset wrapping over a pickle serialized (.bin) file provided by InsightFace Repo.

Parameters:

• name (str.) – Name of val dataset.
• root (str.) – Path to face folder. Default is ‘$(HOME)/mxnet/datasets/face’
• transform (callable, default None.) –

  A function that takes data and transforms them:

  transform = lambda data: data.astype(np.float32)/255