MIC-DKFZ / medicaldetectiontoolkit

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How can I only evaluate based on the ground-truth bounding box?

sunformoon opened this issue · comments

Thanks for the repo!
I am wondering how can I get the evaluation performance (mainly the classification results) given the ground-truth bounding box?

To clarify, my major task is the classification task, so I would like to obtain the classification results based on the given ground-truth bounding boxes. However, I trained with the detection + classification + segmentation label, in which case I have much more voxel-level annotation to prevent the model from overfitting since I don't have many CT cases.

@pfjaeger Any comments would be greatly appreciated!