!2869 Implemented training and evaluation of simple baselines on a GPU
Merge pull request !2869 from Marina Molchanova/simple_baseline_gpu
Showing
- research/cv/simple_baselines/README.md 370 additions, 0 deletionsresearch/cv/simple_baselines/README.md
- research/cv/simple_baselines/README_CN.md 70 additions, 40 deletionsresearch/cv/simple_baselines/README_CN.md
- research/cv/simple_baselines/eval.py 17 additions, 18 deletionsresearch/cv/simple_baselines/eval.py
- research/cv/simple_baselines/export.py 1 addition, 3 deletionsresearch/cv/simple_baselines/export.py
- research/cv/simple_baselines/postprocess.py 4 additions, 4 deletionsresearch/cv/simple_baselines/postprocess.py
- research/cv/simple_baselines/preprocess.py 4 additions, 4 deletionsresearch/cv/simple_baselines/preprocess.py
- research/cv/simple_baselines/scripts/run_distribute_train_gpu.sh 42 additions, 0 deletions...h/cv/simple_baselines/scripts/run_distribute_train_gpu.sh
- research/cv/simple_baselines/scripts/run_eval_gpu.sh 38 additions, 0 deletionsresearch/cv/simple_baselines/scripts/run_eval_gpu.sh
- research/cv/simple_baselines/scripts/run_standalone_train_gpu.sh 44 additions, 0 deletions...h/cv/simple_baselines/scripts/run_standalone_train_gpu.sh
- research/cv/simple_baselines/src/config.py 12 additions, 13 deletionsresearch/cv/simple_baselines/src/config.py
- research/cv/simple_baselines/src/dataset.py 15 additions, 22 deletionsresearch/cv/simple_baselines/src/dataset.py
- research/cv/simple_baselines/src/network_with_loss.py 5 additions, 9 deletionsresearch/cv/simple_baselines/src/network_with_loss.py
- research/cv/simple_baselines/src/pose_resnet.py 15 additions, 33 deletionsresearch/cv/simple_baselines/src/pose_resnet.py
- research/cv/simple_baselines/src/predict.py 8 additions, 9 deletionsresearch/cv/simple_baselines/src/predict.py
- research/cv/simple_baselines/src/utils/coco.py 3 additions, 12 deletionsresearch/cv/simple_baselines/src/utils/coco.py
- research/cv/simple_baselines/src/utils/nms.py 3 additions, 6 deletionsresearch/cv/simple_baselines/src/utils/nms.py
- research/cv/simple_baselines/src/utils/transforms.py 5 additions, 15 deletionsresearch/cv/simple_baselines/src/utils/transforms.py
- research/cv/simple_baselines/train.py 45 additions, 45 deletionsresearch/cv/simple_baselines/train.py
Please register or sign in to comment