hf-transformers-bot's picture
Update tiny models for SuperGlueForKeypointMatching
ab22260 verified
Raw
History Blame Contribute Delete
849 Bytes
{
"architectures": [
"SuperGlueForKeypointMatching"
],
"attention_probs_dropout_prob": 0.0,
"dtype": "float32",
"gnn_layers_types": [
"self",
"cross",
"self",
"cross"
],
"hidden_size": 64,
"initializer_range": 0.02,
"is_decoder": false,
"keypoint_detector_config": {
"border_removal_distance": 4,
"decoder_hidden_size": 64,
"descriptor_decoder_dim": 64,
"encoder_hidden_sizes": [
32,
64
],
"initializer_range": 0.02,
"keypoint_decoder_dim": 65,
"keypoint_threshold": 0.005,
"max_keypoints": 256,
"model_type": "superpoint",
"nms_radius": 4
},
"keypoint_encoder_sizes": [
32,
64
],
"matching_threshold": 0.2,
"model_type": "superglue",
"num_attention_heads": 4,
"sinkhorn_iterations": 100,
"transformers_version": "5.16.0.dev0"
}