├── .gitattributes ├── .gitignore ├── 4_gpu.json ├── LICENSE ├── README.md ├── app_gradio.py ├── assets ├── method.png └── results.png ├── data_scripts └── cal_bbox_by_seg.py ├── datasets.py ├── inference_global.py ├── inference_global.sh ├── inference_local.py ├── inference_local.sh ├── requirements.txt ├── test_datasets ├── 1.jpg ├── 10.jpg ├── 10_bg.png ├── 11.jpg ├── 11_bg.png ├── 15.jpg ├── 15_bg.png ├── 16.jpg ├── 16_bg.png ├── 17.jpg ├── 17_bg.png ├── 1_bg.png ├── 2.jpg ├── 20.jpg ├── 20_bg.png ├── 2_bg.png ├── 3.jpg ├── 3_bg.png ├── 4.png ├── 4_bg.png ├── 7.jpg └── 7_bg.png ├── train_global.py ├── train_global.sh ├── train_local.py └── train_local.sh /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/.gitignore -------------------------------------------------------------------------------- /4_gpu.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/4_gpu.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/README.md -------------------------------------------------------------------------------- /app_gradio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/app_gradio.py -------------------------------------------------------------------------------- /assets/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/assets/method.png -------------------------------------------------------------------------------- /assets/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/assets/results.png -------------------------------------------------------------------------------- /data_scripts/cal_bbox_by_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/data_scripts/cal_bbox_by_seg.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/datasets.py -------------------------------------------------------------------------------- /inference_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/inference_global.py -------------------------------------------------------------------------------- /inference_global.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/inference_global.sh -------------------------------------------------------------------------------- /inference_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/inference_local.py -------------------------------------------------------------------------------- /inference_local.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/inference_local.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/requirements.txt -------------------------------------------------------------------------------- /test_datasets/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/1.jpg -------------------------------------------------------------------------------- /test_datasets/10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/10.jpg -------------------------------------------------------------------------------- /test_datasets/10_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/10_bg.png -------------------------------------------------------------------------------- /test_datasets/11.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/11.jpg -------------------------------------------------------------------------------- /test_datasets/11_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/11_bg.png -------------------------------------------------------------------------------- /test_datasets/15.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/15.jpg -------------------------------------------------------------------------------- /test_datasets/15_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/15_bg.png -------------------------------------------------------------------------------- /test_datasets/16.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/16.jpg -------------------------------------------------------------------------------- /test_datasets/16_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/16_bg.png -------------------------------------------------------------------------------- /test_datasets/17.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/17.jpg -------------------------------------------------------------------------------- /test_datasets/17_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/17_bg.png -------------------------------------------------------------------------------- /test_datasets/1_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/1_bg.png -------------------------------------------------------------------------------- /test_datasets/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/2.jpg -------------------------------------------------------------------------------- /test_datasets/20.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/20.jpg -------------------------------------------------------------------------------- /test_datasets/20_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/20_bg.png -------------------------------------------------------------------------------- /test_datasets/2_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/2_bg.png -------------------------------------------------------------------------------- /test_datasets/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/3.jpg -------------------------------------------------------------------------------- /test_datasets/3_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/3_bg.png -------------------------------------------------------------------------------- /test_datasets/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/4.png -------------------------------------------------------------------------------- /test_datasets/4_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/4_bg.png -------------------------------------------------------------------------------- /test_datasets/7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/7.jpg -------------------------------------------------------------------------------- /test_datasets/7_bg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/test_datasets/7_bg.png -------------------------------------------------------------------------------- /train_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/train_global.py -------------------------------------------------------------------------------- /train_global.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/train_global.sh -------------------------------------------------------------------------------- /train_local.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/train_local.py -------------------------------------------------------------------------------- /train_local.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/csyxwei/ELITE/HEAD/train_local.sh --------------------------------------------------------------------------------