How to train my own Dataset?

[zhang295498]

This code is well, but how it can do to train my own dataset?what i mean is that what and where should i modify?

[wx-1809]

The total number of categories in the dataset, the total number of training data, the total number of validation sets, these need to be modified, and the rest do not need to be modified. In addition, the format of the dataset you train is tfrecord.