Train on myown data

Hello. Thanks for sharing the code.
I want to train the model with my data. Instead of 20 Pascal classes, my data has only 1 class and it is binary classification. Should I just use a split? What parts of the code should I change?
Looking forward to hearing you.