search

huawei-noah / noah-research

Noah Research
865 stars 161 forks source link

The training environment of TNT #35

Open QY1994-0919 opened 3 years ago

QY1994-0919 commented 3 years ago

Hello, I would like to ask where is the training environment and complete code of TNT project? Is there a PyTorch version?

iamhankai commented 3 years ago

Thanks for you attention. Please refer to https://github.com/huawei-noah/CV-Backbones/tree/master/tnt_pytorch for training.

QY1994-0919 commented 3 years ago

@iamhankai OK, thank you very much. If I want to embed the TNT module in YOlO, do I just need to refer to the file tnt.py?

iamhankai commented 3 years ago

If you need pretrained weights, you can use this: https://www.github.com/rwightman/pytorch-image-models/tree/master/timm/models/tnt.py