_classes = 2 # wheat or not(background) # get number of input features for the classifier in_features = model.roi_heads.box_predictor.cls_score.in_features # replace the pre-trained model's head with a new one model.roi_heads.box_predictor = FastRCNNPredictor(in_features, num_classes...
After training on the long-tailed datasets, we evaluate the models onthe corresponding balanced test/validation datasetsand report the commonly used top-1 accuracy over all classes, denoted as All. To better examine performance variations across classes with different number of examples seen during tr...