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.
Apache License 2.0
1.31k stars 297 forks source link

fix indexerror for retinanet in torch1.x #135

Open alibool opened 2 years ago

alibool commented 2 years ago

keep_ix / fg_probs.shape[1] returns a float tensor which can not be used as index with torch 1.x.