how to apply your code on other dataset?

[chenslcool]

Hi, thanks for your nice job! I want to apply your code on other dataset (such as fire detection), however, in this case, num of case is only 1 (20 for voc and 80 for coco). So, should I remove the loss part for classification(because there exists loss for object confidence) , or just change number of class to 1?