├── .gitattributes ├── ImgForReadme ├── none └── result.jpg ├── README.md ├── _init_paths.py ├── cfgs ├── snet.yml ├── snet_535.yml ├── snet_dla.yml ├── snet_dla_nwpu.yml ├── snet_dla_vedai.yml ├── snet_dla_visdrone.yml └── snet_ls.yml ├── demo.py ├── lib ├── __init__.py ├── build │ ├── lib.linux-x86_64-3.6 │ │ └── model │ │ │ └── _C.cpython-36m-x86_64-linux-gnu.so │ └── temp.linux-x86_64-3.6 │ │ └── home │ │ └── omnisky │ │ └── yyf │ │ └── Thundernet_Pytorch-master │ │ └── lib │ │ └── model │ │ └── csrc │ │ ├── cpu │ │ ├── ROIAlign_cpu.o │ │ └── nms_cpu.o │ │ ├── cuda │ │ ├── ROIAlign_cuda.o │ │ ├── ROIPool_cuda.o │ │ └── nms.o │ │ └── vision.o ├── datasets │ ├── VOCdevkit-matlab-wrapper │ │ ├── get_voc_opts.m │ │ ├── voc_eval.m │ │ └── xVOCap.m │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── car_3class.cpython-36.pyc │ │ ├── car_3class_eval.cpython-36.pyc │ │ ├── coco.cpython-36.pyc │ │ ├── ds_utils.cpython-36.pyc │ │ ├── factory.cpython-36.pyc │ │ ├── imdb.cpython-36.pyc │ │ ├── nwpu_coco.cpython-36.pyc │ │ ├── pascal_voc.cpython-36.pyc │ │ ├── vedai_coco.cpython-36.pyc │ │ ├── visdrone.cpython-36.pyc │ │ ├── visdrone_coco.cpython-36.pyc │ │ ├── visdrone_eval.cpython-36.pyc │ │ └── voc_eval.cpython-36.pyc │ ├── car_3class.py │ ├── car_3class_eval.py │ ├── coco.py │ ├── ds_utils.py │ ├── factory.py │ ├── imagenet.py │ ├── imdb.py │ ├── nwpu_coco.py │ ├── pascal_voc.py │ ├── pascal_voc_rbg.py │ ├── tools │ │ └── mcg_munge.py │ ├── vedai_coco.py │ ├── vg.py │ ├── vg_eval.py │ ├── visdrone.py │ ├── visdrone_coco.py │ ├── visdrone_eval.py │ └── voc_eval.py ├── external │ ├── .gitignore │ ├── Makefile │ ├── __init__.py │ ├── __pycache__ │ │ └── __init__.cpython-36.pyc │ ├── build │ │ └── temp.linux-x86_64-3.6 │ │ │ └── nms.o │ ├── cpu_nms │ │ ├── build │ │ │ ├── lib.linux-x86_64-3.6 │ │ │ │ └── cpu_nms.cpython-36m-x86_64-linux-gnu.so │ │ │ └── temp.linux-x86_64-3.6 │ │ │ │ └── cpu_nms.o │ │ ├── cpu_nms.c │ │ ├── cpu_nms.cpython-36m-x86_64-linux-gnu.so │ │ ├── cpu_nms.pyx │ │ └── setup.py │ ├── gpu_nms │ │ ├── build │ │ │ └── temp.linux-x86_64-3.6 │ │ │ │ └── nms │ │ │ │ ├── gpu_nms.o │ │ │ │ └── nms_kernel.o │ │ ├── nms │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ └── __init__.cpython-36.pyc │ │ │ ├── gpu_nms.cpp │ │ │ ├── gpu_nms.cpython-36m-x86_64-linux-gnu.so │ │ │ ├── gpu_nms.hpp │ │ │ ├── gpu_nms.pyx │ │ │ └── nms_kernel.cu │ │ └── setup.py │ ├── nms.pyx │ └── setup.py ├── model │ ├── _C.cpython-36m-x86_64-linux-gnu.so │ ├── csrc │ │ ├── ROIAlign.h │ │ ├── ROIPool.h │ │ ├── cpu │ │ │ ├── ROIAlign_cpu.cpp │ │ │ ├── nms_cpu.cpp │ │ │ └── vision.h │ │ ├── cuda │ │ │ ├── ROIAlign_cuda.cu │ │ │ ├── ROIPool_cuda.cu │ │ │ ├── nms.cu │ │ │ └── vision.h │ │ ├── nms.h │ │ └── vision.cpp │ ├── faster_rcnn │ │ ├── Snet.py │ │ ├── Snet_dla.py │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── Snet.cpython-36.pyc │ │ │ ├── Snet_dla.cpython-36.pyc │ │ │ ├── __init__.cpython-36.pyc │ │ │ ├── dla.cpython-36.pyc │ │ │ ├── dla_up.cpython-36.pyc │ │ │ ├── faster_rcnn.cpython-36.pyc │ │ │ └── modules.cpython-36.pyc │ │ ├── dla.py │ │ ├── dla_up.py │ │ ├── faster_rcnn.py │ │ ├── modules.py │ │ └── pose_dla_dcn.py │ ├── loss │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ └── losses.cpython-36.pyc │ │ └── losses.py │ ├── rpn │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ ├── anchor_target_layer.cpython-36.pyc │ │ │ ├── bbox_transform.cpython-36.pyc │ │ │ ├── generate_anchors.cpython-36.pyc │ │ │ ├── proposal_layer.cpython-36.pyc │ │ │ ├── proposal_target_layer_cascade.cpython-36.pyc │ │ │ └── rpn.cpython-36.pyc │ │ ├── anchor_target_layer.py │ │ ├── bbox_transform.py │ │ ├── centernet_rpn.py │ │ ├── generate_anchors.py │ │ ├── proposal_layer.py │ │ ├── proposal_target_layer_cascade.py │ │ └── rpn.py │ └── utils │ │ ├── .gitignore │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── blob.cpython-36.pyc │ │ ├── config.cpython-36.pyc │ │ └── net_utils.cpython-36.pyc │ │ ├── bbox.pyx │ │ ├── blob.py │ │ ├── cente_decode.py │ │ ├── config.py │ │ ├── layer_utils.py │ │ ├── logger.py │ │ └── net_utils.py ├── psroialign │ ├── PSROIAlign │ │ ├── .gitattributes │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ └── __init__.cpython-36.pyc │ │ ├── build.sh │ │ ├── build │ │ │ ├── lib.linux-x86_64-3.6 │ │ │ │ └── model │ │ │ │ │ └── _C.cpython-36m-x86_64-linux-gnu.so │ │ │ └── temp.linux-x86_64-3.6 │ │ │ │ └── home │ │ │ │ └── omnisky │ │ │ │ └── yyf │ │ │ │ └── Thundernet_Pytorch-master │ │ │ │ └── lib │ │ │ │ └── psroialign │ │ │ │ └── PSROIAlign │ │ │ │ └── model │ │ │ │ └── csrc │ │ │ │ ├── cuda │ │ │ │ ├── PSROIAlign_cuda.o │ │ │ │ └── PSROIPool_cuda.o │ │ │ │ └── vision.o │ │ ├── model │ │ │ ├── _C.cpython-36m-x86_64-linux-gnu.so │ │ │ ├── __init__.py │ │ │ ├── __pycache__ │ │ │ │ └── __init__.cpython-36.pyc │ │ │ ├── csrc │ │ │ │ ├── .DS_Store │ │ │ │ ├── PSROIAlign.h │ │ │ │ ├── PSROIPool.h │ │ │ │ ├── cuda │ │ │ │ │ ├── PSROIAlign_cuda.cpp │ │ │ │ │ ├── PSROIAlign_cuda.cu │ │ │ │ │ ├── PSROIPool_cuda.cu │ │ │ │ │ └── vision.h │ │ │ │ └── vision.cpp │ │ │ ├── example.py │ │ │ └── roi_layers │ │ │ │ ├── __init__.py │ │ │ │ ├── __pycache__ │ │ │ │ ├── __init__.cpython-36.pyc │ │ │ │ ├── ps_roi_align.cpython-36.pyc │ │ │ │ └── ps_roi_pool.cpython-36.pyc │ │ │ │ ├── ps_roi_align.py │ │ │ │ └── ps_roi_pool.py │ │ └── setup.py │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ └── psroialign.cpython-36.pyc │ ├── pollers.py │ └── psroialign.py ├── roi_data_layer │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── augmentation.cpython-36.pyc │ │ ├── roibatchLoader.cpython-36.pyc │ │ ├── roidb.cpython-36.pyc │ │ └── utils.cpython-36.pyc │ ├── augmentation.py │ ├── minibatch.py │ ├── roibatchLoader.py │ ├── roidb.py │ └── utils.py └── setup.py ├── onnx ├── __init__.py ├── onnx_infer.py ├── rcnn_head_to_onnx.py ├── rpn_to_onnx.py └── utils.py ├── script ├── dataset_split.py ├── dataset_txt.py ├── delete.py ├── demo_car_3class_146.sh ├── demo_car_3class_dla.sh ├── demo_nwpu_dla.sh ├── demo_vedai_dla.sh ├── demo_visdrone_146.sh ├── demo_visdrone_535.sh ├── demo_visdrone_dla.sh ├── demo_voc.sh ├── train_49.sh ├── train_car_3class_146.sh ├── train_car_3class_535.sh ├── train_car_3class_dla.sh ├── train_coco_146.sh ├── train_coco_dla.sh ├── train_nwpu_dla.sh ├── train_vedai_dla.sh ├── train_visdrone_146.sh ├── train_visdrone_535.sh ├── train_visdrone_dla.sh ├── train_voc_146.sh ├── txt2json_nwpu.py ├── txt2json_vedai.py └── txt2json_visdrone.py ├── test_net.py ├── trainval_net.py └── utils.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/.gitattributes -------------------------------------------------------------------------------- /ImgForReadme/none: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /ImgForReadme/result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/ImgForReadme/result.jpg -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/README.md -------------------------------------------------------------------------------- /_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/_init_paths.py -------------------------------------------------------------------------------- /cfgs/snet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet.yml -------------------------------------------------------------------------------- /cfgs/snet_535.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_535.yml -------------------------------------------------------------------------------- /cfgs/snet_dla.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_dla.yml -------------------------------------------------------------------------------- /cfgs/snet_dla_nwpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_dla_nwpu.yml -------------------------------------------------------------------------------- /cfgs/snet_dla_vedai.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_dla_vedai.yml -------------------------------------------------------------------------------- /cfgs/snet_dla_visdrone.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_dla_visdrone.yml -------------------------------------------------------------------------------- /cfgs/snet_ls.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/cfgs/snet_ls.yml -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/demo.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/build/lib.linux-x86_64-3.6/model/_C.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/lib.linux-x86_64-3.6/model/_C.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cpu/ROIAlign_cpu.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cpu/ROIAlign_cpu.o -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cpu/nms_cpu.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cpu/nms_cpu.o -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/ROIAlign_cuda.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/ROIAlign_cuda.o -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/ROIPool_cuda.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/ROIPool_cuda.o -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/nms.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/cuda/nms.o -------------------------------------------------------------------------------- /lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/vision.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/model/csrc/vision.o -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/voc_eval.m -------------------------------------------------------------------------------- /lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/VOCdevkit-matlab-wrapper/xVOCap.m -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__init__.py -------------------------------------------------------------------------------- /lib/datasets/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/car_3class.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/car_3class.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/car_3class_eval.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/car_3class_eval.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/coco.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/ds_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/ds_utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/factory.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/factory.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/imdb.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/imdb.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/nwpu_coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/nwpu_coco.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/pascal_voc.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/pascal_voc.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/vedai_coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/vedai_coco.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/visdrone.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/visdrone.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/visdrone_coco.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/visdrone_coco.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/visdrone_eval.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/visdrone_eval.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/__pycache__/voc_eval.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/__pycache__/voc_eval.cpython-36.pyc -------------------------------------------------------------------------------- /lib/datasets/car_3class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/car_3class.py -------------------------------------------------------------------------------- /lib/datasets/car_3class_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/car_3class_eval.py -------------------------------------------------------------------------------- /lib/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/coco.py -------------------------------------------------------------------------------- /lib/datasets/ds_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/ds_utils.py -------------------------------------------------------------------------------- /lib/datasets/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/factory.py -------------------------------------------------------------------------------- /lib/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/imagenet.py -------------------------------------------------------------------------------- /lib/datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/imdb.py -------------------------------------------------------------------------------- /lib/datasets/nwpu_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/nwpu_coco.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/pascal_voc.py -------------------------------------------------------------------------------- /lib/datasets/pascal_voc_rbg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/pascal_voc_rbg.py -------------------------------------------------------------------------------- /lib/datasets/tools/mcg_munge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/tools/mcg_munge.py -------------------------------------------------------------------------------- /lib/datasets/vedai_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/vedai_coco.py -------------------------------------------------------------------------------- /lib/datasets/vg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/vg.py -------------------------------------------------------------------------------- /lib/datasets/vg_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/vg_eval.py -------------------------------------------------------------------------------- /lib/datasets/visdrone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/visdrone.py -------------------------------------------------------------------------------- /lib/datasets/visdrone_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/visdrone_coco.py -------------------------------------------------------------------------------- /lib/datasets/visdrone_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/visdrone_eval.py -------------------------------------------------------------------------------- /lib/datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/datasets/voc_eval.py -------------------------------------------------------------------------------- /lib/external/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/.gitignore -------------------------------------------------------------------------------- /lib/external/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/Makefile -------------------------------------------------------------------------------- /lib/external/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/external/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/external/build/temp.linux-x86_64-3.6/nms.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/build/temp.linux-x86_64-3.6/nms.o -------------------------------------------------------------------------------- /lib/external/cpu_nms/build/lib.linux-x86_64-3.6/cpu_nms.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/build/lib.linux-x86_64-3.6/cpu_nms.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/external/cpu_nms/build/temp.linux-x86_64-3.6/cpu_nms.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/build/temp.linux-x86_64-3.6/cpu_nms.o -------------------------------------------------------------------------------- /lib/external/cpu_nms/cpu_nms.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/cpu_nms.c -------------------------------------------------------------------------------- /lib/external/cpu_nms/cpu_nms.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/cpu_nms.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/external/cpu_nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/cpu_nms.pyx -------------------------------------------------------------------------------- /lib/external/cpu_nms/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/cpu_nms/setup.py -------------------------------------------------------------------------------- /lib/external/gpu_nms/build/temp.linux-x86_64-3.6/nms/gpu_nms.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/build/temp.linux-x86_64-3.6/nms/gpu_nms.o -------------------------------------------------------------------------------- /lib/external/gpu_nms/build/temp.linux-x86_64-3.6/nms/nms_kernel.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/build/temp.linux-x86_64-3.6/nms/nms_kernel.o -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/gpu_nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/gpu_nms.cpp -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/gpu_nms.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /lib/external/gpu_nms/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/nms/nms_kernel.cu -------------------------------------------------------------------------------- /lib/external/gpu_nms/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/gpu_nms/setup.py -------------------------------------------------------------------------------- /lib/external/nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/nms.pyx -------------------------------------------------------------------------------- /lib/external/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/external/setup.py -------------------------------------------------------------------------------- /lib/model/_C.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/_C.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/model/csrc/ROIAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/ROIAlign.h -------------------------------------------------------------------------------- /lib/model/csrc/ROIPool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/ROIPool.h -------------------------------------------------------------------------------- /lib/model/csrc/cpu/ROIAlign_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cpu/ROIAlign_cpu.cpp -------------------------------------------------------------------------------- /lib/model/csrc/cpu/nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cpu/nms_cpu.cpp -------------------------------------------------------------------------------- /lib/model/csrc/cpu/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cpu/vision.h -------------------------------------------------------------------------------- /lib/model/csrc/cuda/ROIAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cuda/ROIAlign_cuda.cu -------------------------------------------------------------------------------- /lib/model/csrc/cuda/ROIPool_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cuda/ROIPool_cuda.cu -------------------------------------------------------------------------------- /lib/model/csrc/cuda/nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cuda/nms.cu -------------------------------------------------------------------------------- /lib/model/csrc/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/cuda/vision.h -------------------------------------------------------------------------------- /lib/model/csrc/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/nms.h -------------------------------------------------------------------------------- /lib/model/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/csrc/vision.cpp -------------------------------------------------------------------------------- /lib/model/faster_rcnn/Snet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/Snet.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/Snet_dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/Snet_dla.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/Snet.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/Snet.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/Snet_dla.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/Snet_dla.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/dla.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/dla.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/dla_up.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/dla_up.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/faster_rcnn.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/faster_rcnn.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/__pycache__/modules.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/__pycache__/modules.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/faster_rcnn/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/dla.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/dla_up.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/dla_up.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/faster_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/faster_rcnn.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/modules.py -------------------------------------------------------------------------------- /lib/model/faster_rcnn/pose_dla_dcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/faster_rcnn/pose_dla_dcn.py -------------------------------------------------------------------------------- /lib/model/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/model/loss/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/loss/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/loss/__pycache__/losses.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/loss/__pycache__/losses.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/loss/losses.py -------------------------------------------------------------------------------- /lib/model/rpn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/anchor_target_layer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/anchor_target_layer.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/bbox_transform.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/bbox_transform.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/generate_anchors.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/generate_anchors.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/proposal_layer.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/proposal_layer.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/proposal_target_layer_cascade.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/proposal_target_layer_cascade.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/__pycache__/rpn.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/__pycache__/rpn.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/rpn/anchor_target_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/anchor_target_layer.py -------------------------------------------------------------------------------- /lib/model/rpn/bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/bbox_transform.py -------------------------------------------------------------------------------- /lib/model/rpn/centernet_rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/centernet_rpn.py -------------------------------------------------------------------------------- /lib/model/rpn/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/generate_anchors.py -------------------------------------------------------------------------------- /lib/model/rpn/proposal_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/proposal_layer.py -------------------------------------------------------------------------------- /lib/model/rpn/proposal_target_layer_cascade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/proposal_target_layer_cascade.py -------------------------------------------------------------------------------- /lib/model/rpn/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/rpn/rpn.py -------------------------------------------------------------------------------- /lib/model/utils/.gitignore: -------------------------------------------------------------------------------- 1 | *.c 2 | *.cpp 3 | *.so 4 | -------------------------------------------------------------------------------- /lib/model/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/model/utils/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/utils/__pycache__/blob.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/__pycache__/blob.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/utils/__pycache__/config.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/__pycache__/config.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/utils/__pycache__/net_utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/__pycache__/net_utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/model/utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/bbox.pyx -------------------------------------------------------------------------------- /lib/model/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/blob.py -------------------------------------------------------------------------------- /lib/model/utils/cente_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/cente_decode.py -------------------------------------------------------------------------------- /lib/model/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/config.py -------------------------------------------------------------------------------- /lib/model/utils/layer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/layer_utils.py -------------------------------------------------------------------------------- /lib/model/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/logger.py -------------------------------------------------------------------------------- /lib/model/utils/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/model/utils/net_utils.py -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/.gitattributes -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/LICENSE -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/README.md -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/build.sh -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/build/lib.linux-x86_64-3.6/model/_C.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/build/lib.linux-x86_64-3.6/model/_C.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.o -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIPool_cuda.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIPool_cuda.o -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/vision.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/build/temp.linux-x86_64-3.6/home/omnisky/yyf/Thundernet_Pytorch-master/lib/psroialign/PSROIAlign/model/csrc/vision.o -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/_C.cpython-36m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/_C.cpython-36m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/.DS_Store -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/PSROIAlign.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/PSROIAlign.h -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/PSROIPool.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/PSROIPool.h -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.cpp -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIAlign_cuda.cu -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIPool_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/cuda/PSROIPool_cuda.cu -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/cuda/vision.h -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/csrc/vision.cpp -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/example.py -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/__init__.py -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/ps_roi_align.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/ps_roi_align.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/ps_roi_pool.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/__pycache__/ps_roi_pool.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/ps_roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/ps_roi_align.py -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/model/roi_layers/ps_roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/model/roi_layers/ps_roi_pool.py -------------------------------------------------------------------------------- /lib/psroialign/PSROIAlign/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/PSROIAlign/setup.py -------------------------------------------------------------------------------- /lib/psroialign/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/psroialign/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/__pycache__/psroialign.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/__pycache__/psroialign.cpython-36.pyc -------------------------------------------------------------------------------- /lib/psroialign/pollers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/pollers.py -------------------------------------------------------------------------------- /lib/psroialign/psroialign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/psroialign/psroialign.py -------------------------------------------------------------------------------- /lib/roi_data_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__init__.py -------------------------------------------------------------------------------- /lib/roi_data_layer/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /lib/roi_data_layer/__pycache__/augmentation.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__pycache__/augmentation.cpython-36.pyc -------------------------------------------------------------------------------- /lib/roi_data_layer/__pycache__/roibatchLoader.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__pycache__/roibatchLoader.cpython-36.pyc -------------------------------------------------------------------------------- /lib/roi_data_layer/__pycache__/roidb.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__pycache__/roidb.cpython-36.pyc -------------------------------------------------------------------------------- /lib/roi_data_layer/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /lib/roi_data_layer/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/augmentation.py -------------------------------------------------------------------------------- /lib/roi_data_layer/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/minibatch.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roibatchLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/roibatchLoader.py -------------------------------------------------------------------------------- /lib/roi_data_layer/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/roidb.py -------------------------------------------------------------------------------- /lib/roi_data_layer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/roi_data_layer/utils.py -------------------------------------------------------------------------------- /lib/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/lib/setup.py -------------------------------------------------------------------------------- /onnx/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /onnx/onnx_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/onnx/onnx_infer.py -------------------------------------------------------------------------------- /onnx/rcnn_head_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/onnx/rcnn_head_to_onnx.py -------------------------------------------------------------------------------- /onnx/rpn_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/onnx/rpn_to_onnx.py -------------------------------------------------------------------------------- /onnx/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/onnx/utils.py -------------------------------------------------------------------------------- /script/dataset_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/dataset_split.py -------------------------------------------------------------------------------- /script/dataset_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/dataset_txt.py -------------------------------------------------------------------------------- /script/delete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/delete.py -------------------------------------------------------------------------------- /script/demo_car_3class_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_car_3class_146.sh -------------------------------------------------------------------------------- /script/demo_car_3class_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_car_3class_dla.sh -------------------------------------------------------------------------------- /script/demo_nwpu_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_nwpu_dla.sh -------------------------------------------------------------------------------- /script/demo_vedai_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_vedai_dla.sh -------------------------------------------------------------------------------- /script/demo_visdrone_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_visdrone_146.sh -------------------------------------------------------------------------------- /script/demo_visdrone_535.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_visdrone_535.sh -------------------------------------------------------------------------------- /script/demo_visdrone_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_visdrone_dla.sh -------------------------------------------------------------------------------- /script/demo_voc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/demo_voc.sh -------------------------------------------------------------------------------- /script/train_49.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_49.sh -------------------------------------------------------------------------------- /script/train_car_3class_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_car_3class_146.sh -------------------------------------------------------------------------------- /script/train_car_3class_535.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_car_3class_535.sh -------------------------------------------------------------------------------- /script/train_car_3class_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_car_3class_dla.sh -------------------------------------------------------------------------------- /script/train_coco_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_coco_146.sh -------------------------------------------------------------------------------- /script/train_coco_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_coco_dla.sh -------------------------------------------------------------------------------- /script/train_nwpu_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_nwpu_dla.sh -------------------------------------------------------------------------------- /script/train_vedai_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_vedai_dla.sh -------------------------------------------------------------------------------- /script/train_visdrone_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_visdrone_146.sh -------------------------------------------------------------------------------- /script/train_visdrone_535.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_visdrone_535.sh -------------------------------------------------------------------------------- /script/train_visdrone_dla.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_visdrone_dla.sh -------------------------------------------------------------------------------- /script/train_voc_146.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/train_voc_146.sh -------------------------------------------------------------------------------- /script/txt2json_nwpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/txt2json_nwpu.py -------------------------------------------------------------------------------- /script/txt2json_vedai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/txt2json_vedai.py -------------------------------------------------------------------------------- /script/txt2json_visdrone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/script/txt2json_visdrone.py -------------------------------------------------------------------------------- /test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/test_net.py -------------------------------------------------------------------------------- /trainval_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/trainval_net.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yyf7329081/small-object-detection-with-lightweight-network/HEAD/utils.py --------------------------------------------------------------------------------