i want to train only two classes. one class + background , can you pls. tell me how i modify the code ? thanks!