├── C_f.zip ├── README.md ├── WESNet.png ├── config.py ├── data ├── __init__.py └── dataset.py ├── inference.py ├── model ├── __init__.py ├── detector.py ├── loss.py ├── loss_pho.py ├── loss_pro.py └── network.py ├── prepare_dataset.cpp ├── requirements.txt ├── train.py ├── util ├── __init__.py └── log.py └── weights └── WESNet_detector.pth /C_f.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/C_f.zip -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/README.md -------------------------------------------------------------------------------- /WESNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/WESNet.png -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/config.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/data/dataset.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/inference.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/detector.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/loss.py -------------------------------------------------------------------------------- /model/loss_pho.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/loss_pho.py -------------------------------------------------------------------------------- /model/loss_pro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/loss_pro.py -------------------------------------------------------------------------------- /model/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/model/network.py -------------------------------------------------------------------------------- /prepare_dataset.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/prepare_dataset.cpp -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/train.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | """Utility related package.""" 2 | from .log import Logger -------------------------------------------------------------------------------- /util/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/util/log.py -------------------------------------------------------------------------------- /weights/WESNet_detector.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dokidoki-yo/WESNet/HEAD/weights/WESNet_detector.pth --------------------------------------------------------------------------------