!3597 [开源之夏][GPU]StyTr^2模型
Merge pull request !3597 from ym/stytr
No related branches found
No related tags found
Showing
- research/cv/StyTr-2/README_CN.md 272 additions, 0 deletionsresearch/cv/StyTr-2/README_CN.md
- research/cv/StyTr-2/eval.py 142 additions, 0 deletionsresearch/cv/StyTr-2/eval.py
- research/cv/StyTr-2/requirements.txt 3 additions, 0 deletionsresearch/cv/StyTr-2/requirements.txt
- research/cv/StyTr-2/scripts/run_eval.sh 44 additions, 0 deletionsresearch/cv/StyTr-2/scripts/run_eval.sh
- research/cv/StyTr-2/scripts/run_train.sh 42 additions, 0 deletionsresearch/cv/StyTr-2/scripts/run_train.sh
- research/cv/StyTr-2/scripts/run_train_distribute.sh 46 additions, 0 deletionsresearch/cv/StyTr-2/scripts/run_train_distribute.sh
- research/cv/StyTr-2/src/models/StyTR.py 179 additions, 0 deletionsresearch/cv/StyTr-2/src/models/StyTR.py
- research/cv/StyTr-2/src/models/ViT_helper.py 17 additions, 0 deletionsresearch/cv/StyTr-2/src/models/ViT_helper.py
- research/cv/StyTr-2/src/models/WithLossCell.py 92 additions, 0 deletionsresearch/cv/StyTr-2/src/models/WithLossCell.py
- research/cv/StyTr-2/src/models/transformer.py 425 additions, 0 deletionsresearch/cv/StyTr-2/src/models/transformer.py
- research/cv/StyTr-2/src/utils/function.py 59 additions, 0 deletionsresearch/cv/StyTr-2/src/utils/function.py
- research/cv/StyTr-2/src/utils/split_dataset.py 53 additions, 0 deletionsresearch/cv/StyTr-2/src/utils/split_dataset.py
- research/cv/StyTr-2/train.py 229 additions, 0 deletionsresearch/cv/StyTr-2/train.py
Please register or sign in to comment