├── .github └── FUNDING.yml ├── .gitignore ├── LICENSE.md ├── README.md ├── adamp.py ├── compute_loss.py ├── data ├── test │ └── coco_test │ │ ├── image │ │ └── 000000000872.jpg │ │ └── mask │ │ └── 000000000872.png └── train │ └── coco_train │ ├── image │ ├── 000000000785.jpg │ └── 000000001296.jpg │ └── mask │ ├── 000000000785.png │ └── 000000001296.png ├── data_loader.py ├── distributed_train.py ├── inference.py ├── model.py ├── requirements.txt └── webcam.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | .idea/ 3 | 4 | *.pyc -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/README.md -------------------------------------------------------------------------------- /adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/adamp.py -------------------------------------------------------------------------------- /compute_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/compute_loss.py -------------------------------------------------------------------------------- /data/test/coco_test/image/000000000872.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/test/coco_test/image/000000000872.jpg -------------------------------------------------------------------------------- /data/test/coco_test/mask/000000000872.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/test/coco_test/mask/000000000872.png -------------------------------------------------------------------------------- /data/train/coco_train/image/000000000785.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/train/coco_train/image/000000000785.jpg -------------------------------------------------------------------------------- /data/train/coco_train/image/000000001296.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/train/coco_train/image/000000001296.jpg -------------------------------------------------------------------------------- /data/train/coco_train/mask/000000000785.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/train/coco_train/mask/000000000785.png -------------------------------------------------------------------------------- /data/train/coco_train/mask/000000001296.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data/train/coco_train/mask/000000001296.png -------------------------------------------------------------------------------- /data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/data_loader.py -------------------------------------------------------------------------------- /distributed_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/distributed_train.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/inference.py -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | albumentations 2 | numpy 3 | opencv-python 4 | tqdm 5 | tensorflow_gpu>=2.3.1 6 | -------------------------------------------------------------------------------- /webcam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cpuimage/SINet/HEAD/webcam.py --------------------------------------------------------------------------------