├── .gitignore ├── README.md ├── data ├── .gitignore └── example_bgs │ └── tokyo.jpg ├── data_loader.py ├── infer_images.py ├── infer_video.py ├── models ├── __init__.py ├── detectron2.py ├── jitnet.py └── u2net.py ├── requirements.txt └── train.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | saved_models 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/README.md -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | example_videos 2 | out 3 | davis 4 | supervisely 5 | zoom 6 | -------------------------------------------------------------------------------- /data/example_bgs/tokyo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/data/example_bgs/tokyo.jpg -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/data_loader.py -------------------------------------------------------------------------------- /infer_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/infer_images.py -------------------------------------------------------------------------------- /infer_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/infer_video.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/models/detectron2.py -------------------------------------------------------------------------------- /models/jitnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/models/jitnet.py -------------------------------------------------------------------------------- /models/u2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/models/u2net.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/josephch405/jit-masker/HEAD/train.py --------------------------------------------------------------------------------