├── .gitattributes ├── .gitignore ├── README.md ├── __init__.py ├── classes.txt ├── configs ├── __init__.py ├── config_inception_resnet.py ├── config_res101.py ├── config_resnet_50.py └── config_vgg16.py ├── convert_txt2xml.py ├── data ├── __init__.py ├── io │ ├── __init__.py │ ├── convert_data_to_tfrecord.py │ ├── divide_data.py │ ├── image_preprocess.py │ └── read_tfrecord.py ├── pretrained_weights │ └── README.md └── tfrecords │ └── README.md ├── gen_classes.py ├── graph.png ├── help_utils ├── __init__.py ├── help_utils.py └── tools.py ├── libs ├── __init__.py ├── box_utils │ ├── __init__.py │ ├── anchor_utils_pyfunc.py │ ├── boxes_utils.py │ ├── encode_and_decode.py │ ├── iou.py │ ├── make_anchor.py │ ├── nms.py │ ├── show_box_in_tensor.py │ └── visualization.py ├── configs │ ├── __init__.py │ └── cfgs.py ├── fast_rcnn │ ├── __init__.py │ └── build_fast_rcnn.py ├── label_name_dict │ ├── __init__.py │ └── label_dict.py ├── losses │ ├── __init__.py │ └── losses.py ├── networks │ ├── __init__.py │ ├── nets │ │ ├── __init__.py │ │ └── vggnet16.py │ ├── network_factory.py │ └── slim_nets │ │ ├── __init__.py │ │ ├── alexnet.py │ │ ├── alexnet_test.py │ │ ├── cifarnet.py │ │ ├── inception.py │ │ ├── inception_resnet_v2.py │ │ ├── inception_resnet_v2_test.py │ │ ├── inception_utils.py │ │ ├── inception_v1.py │ │ ├── inception_v1_test.py │ │ ├── inception_v2.py │ │ ├── inception_v2_test.py │ │ ├── inception_v3.py │ │ ├── inception_v3_test.py │ │ ├── inception_v4.py │ │ ├── inception_v4_test.py │ │ ├── lenet.py │ │ ├── mobilenet_v1.md │ │ ├── mobilenet_v1.png │ │ ├── mobilenet_v1.py │ │ ├── mobilenet_v1_test.py │ │ ├── nets_factory.py │ │ ├── nets_factory_test.py │ │ ├── overfeat.py │ │ ├── overfeat_test.py │ │ ├── resnet_utils.py │ │ ├── resnet_v1.py │ │ ├── resnet_v1_test.py │ │ ├── resnet_v2.py │ │ ├── resnet_v2_test.py │ │ ├── vgg.py │ │ └── vgg_test.py ├── rpn │ ├── __init__.py │ └── build_rpn.py └── tf_utils │ ├── __init__.py │ └── tf_ops.py ├── scripts ├── demo.sh ├── eval.sh ├── inference.sh ├── test.sh └── train.sh └── tools ├── __init__.py ├── demo.py ├── eval.py ├── inference.py ├── inference_result └── README.md ├── restore_model.py ├── test.py └── train.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /classes.txt: -------------------------------------------------------------------------------- 1 | 层 2 | -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/config_inception_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/configs/config_inception_resnet.py -------------------------------------------------------------------------------- /configs/config_res101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/configs/config_res101.py -------------------------------------------------------------------------------- /configs/config_resnet_50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/configs/config_resnet_50.py -------------------------------------------------------------------------------- /configs/config_vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/configs/config_vgg16.py -------------------------------------------------------------------------------- /convert_txt2xml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/convert_txt2xml.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/io/convert_data_to_tfrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/io/convert_data_to_tfrecord.py -------------------------------------------------------------------------------- /data/io/divide_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/io/divide_data.py -------------------------------------------------------------------------------- /data/io/image_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/io/image_preprocess.py -------------------------------------------------------------------------------- /data/io/read_tfrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/io/read_tfrecord.py -------------------------------------------------------------------------------- /data/pretrained_weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/pretrained_weights/README.md -------------------------------------------------------------------------------- /data/tfrecords/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/data/tfrecords/README.md -------------------------------------------------------------------------------- /gen_classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/gen_classes.py -------------------------------------------------------------------------------- /graph.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/graph.png -------------------------------------------------------------------------------- /help_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /help_utils/help_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/help_utils/help_utils.py -------------------------------------------------------------------------------- /help_utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/help_utils/tools.py -------------------------------------------------------------------------------- /libs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/box_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/box_utils/anchor_utils_pyfunc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/anchor_utils_pyfunc.py -------------------------------------------------------------------------------- /libs/box_utils/boxes_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/boxes_utils.py -------------------------------------------------------------------------------- /libs/box_utils/encode_and_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/encode_and_decode.py -------------------------------------------------------------------------------- /libs/box_utils/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/iou.py -------------------------------------------------------------------------------- /libs/box_utils/make_anchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/make_anchor.py -------------------------------------------------------------------------------- /libs/box_utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/nms.py -------------------------------------------------------------------------------- /libs/box_utils/show_box_in_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/show_box_in_tensor.py -------------------------------------------------------------------------------- /libs/box_utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/box_utils/visualization.py -------------------------------------------------------------------------------- /libs/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/configs/cfgs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/configs/cfgs.py -------------------------------------------------------------------------------- /libs/fast_rcnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/fast_rcnn/build_fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/fast_rcnn/build_fast_rcnn.py -------------------------------------------------------------------------------- /libs/label_name_dict/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/label_name_dict/label_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/label_name_dict/label_dict.py -------------------------------------------------------------------------------- /libs/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/losses/losses.py -------------------------------------------------------------------------------- /libs/networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/networks/nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/networks/nets/vggnet16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/nets/vggnet16.py -------------------------------------------------------------------------------- /libs/networks/network_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/network_factory.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /libs/networks/slim_nets/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/alexnet.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/alexnet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/alexnet_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/cifarnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/cifarnet.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_resnet_v2.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_resnet_v2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_resnet_v2_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_utils.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v1.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v1_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v1_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v2.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v2_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v3.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v3_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v3_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v4.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/inception_v4_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/inception_v4_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/lenet.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/mobilenet_v1.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/mobilenet_v1.md -------------------------------------------------------------------------------- /libs/networks/slim_nets/mobilenet_v1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/mobilenet_v1.png -------------------------------------------------------------------------------- /libs/networks/slim_nets/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/mobilenet_v1.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/mobilenet_v1_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/mobilenet_v1_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/nets_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/nets_factory.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/nets_factory_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/nets_factory_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/overfeat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/overfeat.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/overfeat_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/overfeat_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/resnet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/resnet_utils.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/resnet_v1.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/resnet_v1_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/resnet_v1_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/resnet_v2.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/resnet_v2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/resnet_v2_test.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/vgg.py -------------------------------------------------------------------------------- /libs/networks/slim_nets/vgg_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/networks/slim_nets/vgg_test.py -------------------------------------------------------------------------------- /libs/rpn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/rpn/build_rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/rpn/build_rpn.py -------------------------------------------------------------------------------- /libs/tf_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /libs/tf_utils/tf_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/libs/tf_utils/tf_ops.py -------------------------------------------------------------------------------- /scripts/demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/scripts/demo.sh -------------------------------------------------------------------------------- /scripts/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/scripts/eval.sh -------------------------------------------------------------------------------- /scripts/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/scripts/inference.sh -------------------------------------------------------------------------------- /scripts/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/scripts/test.sh -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/demo.py -------------------------------------------------------------------------------- /tools/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/eval.py -------------------------------------------------------------------------------- /tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/inference.py -------------------------------------------------------------------------------- /tools/inference_result/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/inference_result/README.md -------------------------------------------------------------------------------- /tools/restore_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/restore_model.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kongsea/FPN_TensorFlow/HEAD/tools/train.py --------------------------------------------------------------------------------