├── README.md ├── config └── cod_mgl50.yaml ├── env.yaml ├── lib └── sync_bn │ ├── functions │ ├── __init__.py │ └── sync_bn.py │ ├── modules │ ├── __init__.py │ └── sync_bn.py │ └── src │ ├── __init__.py │ ├── cpu │ ├── operator.cpp │ ├── operator.h │ └── sync_bn.cpp │ └── gpu │ ├── common.h │ ├── device_tensor.h │ ├── operator.cpp │ ├── operator.h │ └── sync_bn_cuda.cu ├── model ├── basicnet.py ├── mglnet.py └── resnet.py ├── pre_trained └── model.txt ├── test.py ├── train.py ├── train_test_file ├── CAMO_test.lst ├── CHAMELEON_test.lst ├── COD10K_test.lst ├── NC4K_test.lst └── train.lst └── util ├── config.py ├── dataset.py ├── transform.py └── util.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/README.md -------------------------------------------------------------------------------- /config/cod_mgl50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/config/cod_mgl50.yaml -------------------------------------------------------------------------------- /env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/env.yaml -------------------------------------------------------------------------------- /lib/sync_bn/functions/__init__.py: -------------------------------------------------------------------------------- 1 | from .sync_bn import * 2 | -------------------------------------------------------------------------------- /lib/sync_bn/functions/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/functions/sync_bn.py -------------------------------------------------------------------------------- /lib/sync_bn/modules/__init__.py: -------------------------------------------------------------------------------- 1 | from .sync_bn import * 2 | -------------------------------------------------------------------------------- /lib/sync_bn/modules/sync_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/modules/sync_bn.py -------------------------------------------------------------------------------- /lib/sync_bn/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/__init__.py -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/cpu/operator.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/cpu/operator.h -------------------------------------------------------------------------------- /lib/sync_bn/src/cpu/sync_bn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/cpu/sync_bn.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/gpu/common.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/device_tensor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/gpu/device_tensor.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/operator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/gpu/operator.cpp -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/operator.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/gpu/operator.h -------------------------------------------------------------------------------- /lib/sync_bn/src/gpu/sync_bn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/lib/sync_bn/src/gpu/sync_bn_cuda.cu -------------------------------------------------------------------------------- /model/basicnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/model/basicnet.py -------------------------------------------------------------------------------- /model/mglnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/model/mglnet.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/model/resnet.py -------------------------------------------------------------------------------- /pre_trained/model.txt: -------------------------------------------------------------------------------- 1 | https://drive.google.com/file/d/17WYyKg40DkAgFWOusiAKgqZOlfUFzjn5/view 2 | -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train.py -------------------------------------------------------------------------------- /train_test_file/CAMO_test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train_test_file/CAMO_test.lst -------------------------------------------------------------------------------- /train_test_file/CHAMELEON_test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train_test_file/CHAMELEON_test.lst -------------------------------------------------------------------------------- /train_test_file/COD10K_test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train_test_file/COD10K_test.lst -------------------------------------------------------------------------------- /train_test_file/NC4K_test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train_test_file/NC4K_test.lst -------------------------------------------------------------------------------- /train_test_file/train.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/train_test_file/train.lst -------------------------------------------------------------------------------- /util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/util/config.py -------------------------------------------------------------------------------- /util/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/util/dataset.py -------------------------------------------------------------------------------- /util/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/util/transform.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fanyang587/MGL/HEAD/util/util.py --------------------------------------------------------------------------------