How can I use my dataset to train the model?

[ST-Jack]

My dataset only have one class, could you share the way to modify the code?

[zbf1991]

Please prepare the *.npy file to encode your class as a one-hot encoding (you can find a *.npy file in this project). Then prepare your train.txt like the "train_aug.txt" in this project. Finally, please prepare your image in VOCdevkit style. After modifying the path as your file path, you can train on your dataset