gouthamvgk / SuperGlue_training

This repository contains the training code for SuperGlue. Uses COCO dataset by generating random homographies.
Other
101 stars 22 forks source link

Training with different descriptions #16

Open antithing opened 1 year ago

antithing commented 1 year ago

Hi, I am trying to train superglue using hfnet keypoints instead of superpoint.

Can you please point me to the code that I need to change for this? Where is superpoint run?

Thanks!

1061700625 commented 6 months ago

文件train_superglue.py。对于superglue来说,只要以下格式正确就行:

superglue_input = {
    'keypoints0': keypoints0, 'keypoints1': keypoints1,
    'descriptors0': descriptors0, 'descriptors1': descriptors1,
    'image0': images0, 'image1': images1,
    'scores0': scores0, 'scores1': scores1,
    'matches': match_indexes,
    'gt_vec': gt_vector,
}