├── IR_ckpt └── info.txt ├── README.md ├── args.yml ├── config ├── dataset └── info.txt ├── models ├── SE_block.py ├── __init__.py ├── __pycache__ │ ├── SE_block.cpython-36.pyc │ ├── __init__.cpython-36.pyc │ ├── __init__.cpython-37.pyc │ ├── csp_darknet.cpython-36.pyc │ ├── csp_darknet.cpython-37.pyc │ ├── csp_resnext.cpython-36.pyc │ ├── csp_resnext.cpython-37.pyc │ ├── darknet.cpython-36.pyc │ ├── darknet.cpython-37.pyc │ ├── densenet.cpython-36.pyc │ ├── densenet.cpython-37.pyc │ ├── inception_blocks.cpython-36.pyc │ ├── mnasnet.cpython-36.pyc │ ├── mnasnet.cpython-37.pyc │ ├── resnet.cpython-36.pyc │ ├── resnet.cpython-37.pyc │ └── resnext.cpython-36.pyc ├── blocks │ ├── SE_block.py │ ├── __init__.py │ ├── __pycache__ │ │ ├── SE_block.cpython-36.pyc │ │ ├── SE_block.cpython-37.pyc │ │ ├── __init__.cpython-36.pyc │ │ ├── __init__.cpython-37.pyc │ │ ├── conv_bn.cpython-36.pyc │ │ ├── conv_bn.cpython-37.pyc │ │ ├── dense_block.cpython-36.pyc │ │ ├── dense_block.cpython-37.pyc │ │ ├── residual_blocks.cpython-36.pyc │ │ └── residual_blocks.cpython-37.pyc │ ├── conv_bn.py │ ├── dense_block.py │ ├── dpn_block.py │ ├── inception_blocks.py │ ├── residual_blocks.py │ ├── resnext_block.py │ └── shuffle_block.py ├── csp_darknet.py ├── csp_resnext.py ├── darknet.py ├── densenet.py ├── dpn.py ├── googlenet.py ├── inception_blocks.py ├── inception_v3.py ├── inception_v4_resnet.py ├── mnasnet.py ├── resnet.py ├── resnext.py ├── shufflenet_v2.py └── vgg.py ├── onnx_ckpt └── info.txt ├── onnx_infer.py ├── predict.py ├── test_net.py ├── to_onnx.py ├── toch_ckpt └── info.txt ├── train.py ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── arg_utils.cpython-36.pyc │ ├── data_utils.cpython-36.pyc │ ├── earlystopping.cpython-36.pyc │ └── progress_utils.cpython-36.pyc ├── arg_utils.py ├── data_utils.py ├── earlystopping.py └── progress_utils.py └── vino_infer.py /IR_ckpt/info.txt: -------------------------------------------------------------------------------- 1 | directory to store checkpoints. -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/README.md -------------------------------------------------------------------------------- /args.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/args.yml -------------------------------------------------------------------------------- /config: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/info.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/dataset/info.txt -------------------------------------------------------------------------------- /models/SE_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/SE_block.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/__pycache__/SE_block.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/SE_block.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/csp_darknet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/csp_darknet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/csp_darknet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/csp_darknet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/csp_resnext.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/csp_resnext.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/csp_resnext.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/csp_resnext.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/darknet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/darknet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/darknet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/darknet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/densenet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/densenet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/densenet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/densenet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/inception_blocks.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/inception_blocks.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/mnasnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/mnasnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/mnasnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/mnasnet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/resnet.cpython-36.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnet.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/resnet.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/resnext.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/__pycache__/resnext.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/SE_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/SE_block.py -------------------------------------------------------------------------------- /models/blocks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/blocks/__pycache__/SE_block.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/SE_block.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/SE_block.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/SE_block.cpython-37.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/conv_bn.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/conv_bn.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/conv_bn.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/conv_bn.cpython-37.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/dense_block.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/dense_block.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/dense_block.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/dense_block.cpython-37.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/residual_blocks.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/residual_blocks.cpython-36.pyc -------------------------------------------------------------------------------- /models/blocks/__pycache__/residual_blocks.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/__pycache__/residual_blocks.cpython-37.pyc -------------------------------------------------------------------------------- /models/blocks/conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/conv_bn.py -------------------------------------------------------------------------------- /models/blocks/dense_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/dense_block.py -------------------------------------------------------------------------------- /models/blocks/dpn_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/dpn_block.py -------------------------------------------------------------------------------- /models/blocks/inception_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/inception_blocks.py -------------------------------------------------------------------------------- /models/blocks/residual_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/residual_blocks.py -------------------------------------------------------------------------------- /models/blocks/resnext_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/resnext_block.py -------------------------------------------------------------------------------- /models/blocks/shuffle_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/blocks/shuffle_block.py -------------------------------------------------------------------------------- /models/csp_darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/csp_darknet.py -------------------------------------------------------------------------------- /models/csp_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/csp_resnext.py -------------------------------------------------------------------------------- /models/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/darknet.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/dpn.py -------------------------------------------------------------------------------- /models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/googlenet.py -------------------------------------------------------------------------------- /models/inception_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/inception_blocks.py -------------------------------------------------------------------------------- /models/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/inception_v3.py -------------------------------------------------------------------------------- /models/inception_v4_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/inception_v4_resnet.py -------------------------------------------------------------------------------- /models/mnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/mnasnet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/resnext.py -------------------------------------------------------------------------------- /models/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/shufflenet_v2.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/models/vgg.py -------------------------------------------------------------------------------- /onnx_ckpt/info.txt: -------------------------------------------------------------------------------- 1 | directory to store checkpoints. -------------------------------------------------------------------------------- /onnx_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/onnx_infer.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/predict.py -------------------------------------------------------------------------------- /test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/test_net.py -------------------------------------------------------------------------------- /to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/to_onnx.py -------------------------------------------------------------------------------- /toch_ckpt/info.txt: -------------------------------------------------------------------------------- 1 | directory to store checkpoints. -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/arg_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/__pycache__/arg_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/data_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/__pycache__/data_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/earlystopping.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/__pycache__/earlystopping.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/progress_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/__pycache__/progress_utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/arg_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/arg_utils.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/earlystopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/earlystopping.py -------------------------------------------------------------------------------- /utils/progress_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/utils/progress_utils.py -------------------------------------------------------------------------------- /vino_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YichaoCai1/torch_backbones/HEAD/vino_infer.py --------------------------------------------------------------------------------