Skip to content

Conversation

Johansmm
Copy link
Contributor

Full train-pipeline to train Detr and Deformable Detr with PanopticHead.

News:

  • DetrPanopticCriterion and DeformablePanopticCriterion, based on PanopticCriterion

Updates:

  • Detr and Deformable forward, to provide the correct backbone format: Input_projection for the last feature of backbone.output
    • In Deformable, backbone was changing to provide the same layers than detr.backbone. Also the encoder.output was provided in deformable_transformer.output.dict .
  • Detr_panoptic forward update to get input_project directly from features[-1][0] and the correct mask format.
  • Threshold boxes in inference given by default threshold values in detr.inference or deformable.inference

Fix other issues:

  • Update ModelCheckpoint to save the top=3 checkpoints (as provided by the argument support)

Closes #98

@Johansmm Johansmm added enhancement New feature or request alonet labels Oct 21, 2021
@Johansmm Johansmm requested a review from thibo73800 October 21, 2021 09:42
@Johansmm Johansmm self-assigned this Oct 21, 2021
@thibo73800 thibo73800 merged commit dd955ce into master Oct 29, 2021
@thibo73800 thibo73800 deleted the 98-deformable-panopticHead branch March 10, 2022 09:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
alonet enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Update PanopticHead to support Deformable DETR
2 participants