DocF / multispectral-object-detection

Multispectral Object Detection with Yolov5 and Transformer
GNU Affero General Public License v3.0
315 stars 60 forks source link

Which part of the code is the cross modality fusion transformer #2

Closed infinite013 closed 3 years ago

infinite013 commented 3 years ago

I can't find the CFT module

DocF commented 3 years ago

U can see the bottom of the ./models/common.py

class GPT(nn.Module):
    """  the full GPT language model, with a context size of block_size """

If u are looking for the structure of the CFT module, U can see

/models/transformer/yolov5l_fusion_transformerx3_FLIR_aligned.yaml
infinite013 commented 3 years ago

thank U