detections = detections[detections.class_id == 1]