| { | |
| "backbone": "resnet18", | |
| "position_embedding": "sine", | |
| "hidden_dim": 384, | |
| "enc_layers": 6, | |
| "dec_layers": 6, | |
| "dim_feedforward": 2048, | |
| "nheads": 8, | |
| "dropout": 0.0, | |
| "pre_norm": false, | |
| "num_queries": 24, | |
| "aux_loss": true, | |
| "num_intra_relation_classes": 10, | |
| "num_inter_relation_classes": 7, | |
| "process_height": 96, | |
| "process_width": 128, | |
| "max_process_window_length": 100, | |
| "weight_file": "OmniShotCut_ckpt.pth" | |
| } | |