How to build a dataset using PyTorch C++ frontend

(Wenbo Ma) #1

I’ve been trying to deploy my model in the form of a desktop application and I’ve successfully loaded my trained model in the C++ frontend. But I don’t know how to build my own dataset using C++ API. The tutorial only demonstrates how to load MNIST dataset. Dose anyone know how to do this?

Thanks in advance.


Creating a C++ dataset is similar to creating a PyTorch python dataset.

One has to subclass the Dataset class, and then implement some base methods.

See the MNIST class for example:

Alternatively, you can subclass from BatchDataset and implement a get_batch method that returns a batch of samples directly:

How can I define my own dataset with the API
(Wenbo Ma) #3

I’ll look into that. Thanks a lot! :innocent: