├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── bin ├── person_tracker.py └── yolov3_deepsort_ros.py ├── configs ├── deep_sort.yaml ├── yolov3.yaml └── yolov3_tiny.yaml ├── launch ├── person_tracker.launch └── person_tracker_srv.launch ├── package.xml ├── resources ├── example.gif └── tracking.gif ├── setup.py ├── src ├── __init__.py ├── deep_sort │ ├── README.md │ ├── __init__.py │ ├── deep │ │ ├── __init__.py │ │ ├── aligned_feature_extractor.py │ │ ├── checkpoint │ │ │ └── .gitkeep │ │ ├── evaluate.py │ │ ├── face_feature_extractor.py │ │ ├── feature_extractor.py │ │ ├── inception_resnet_v1.py │ │ ├── model.py │ │ ├── osnet_ain.py │ │ └── reid_feature_extractor.py │ ├── deep_sort.py │ └── sort │ │ ├── __init__.py │ │ ├── detection.py │ │ ├── iou_matching.py │ │ ├── kalman_filter.py │ │ ├── linear_assignment.py │ │ ├── nn_matching.py │ │ ├── preprocessing.py │ │ ├── track.py │ │ └── tracker.py ├── detector │ ├── YOLOv3 │ │ ├── README.md │ │ ├── __init__.py │ │ ├── cfg.py │ │ ├── cfg │ │ │ ├── coco.data │ │ │ ├── coco.names │ │ │ ├── darknet19_448.cfg │ │ │ ├── tiny-yolo-voc.cfg │ │ │ ├── tiny-yolo.cfg │ │ │ ├── voc.data │ │ │ ├── voc.names │ │ │ ├── voc_gaotie.data │ │ │ ├── yolo-voc.cfg │ │ │ ├── yolo.cfg │ │ │ ├── yolo_v3.cfg │ │ │ └── yolov3-tiny.cfg │ │ ├── darknet.py │ │ ├── demo │ │ │ ├── 004545.jpg │ │ │ └── results │ │ │ │ └── 004545.jpg │ │ ├── detect.py │ │ ├── detector.py │ │ ├── nms │ │ │ ├── __init__.py │ │ │ ├── build.sh │ │ │ ├── ext │ │ │ │ ├── __init__.py │ │ │ │ ├── build.py │ │ │ │ ├── cpu │ │ │ │ │ ├── nms_cpu.cpp │ │ │ │ │ └── vision.h │ │ │ │ ├── cuda │ │ │ │ │ ├── nms.cu │ │ │ │ │ └── vision.h │ │ │ │ ├── nms.h │ │ │ │ └── vision.cpp │ │ │ ├── nms.py │ │ │ └── python_nms.py │ │ ├── region_layer.py │ │ ├── weight │ │ │ └── .gitkeep │ │ ├── yolo_layer.py │ │ └── yolo_utils.py │ └── __init__.py ├── utils │ ├── __init__.py │ ├── configs │ │ ├── deep_sort.yaml │ │ ├── yolov3.yaml │ │ └── yolov3_tiny.yaml │ ├── draw.py │ ├── evaluation.py │ ├── io.py │ ├── log.py │ └── parser.py └── weights │ └── .gitkeep └── srv ├── choose_target.srv ├── clear_target.srv └── update_tracker.srv /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/README.md -------------------------------------------------------------------------------- /bin/person_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/bin/person_tracker.py -------------------------------------------------------------------------------- /bin/yolov3_deepsort_ros.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/bin/yolov3_deepsort_ros.py -------------------------------------------------------------------------------- /configs/deep_sort.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/configs/deep_sort.yaml -------------------------------------------------------------------------------- /configs/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/configs/yolov3.yaml -------------------------------------------------------------------------------- /configs/yolov3_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/configs/yolov3_tiny.yaml -------------------------------------------------------------------------------- /launch/person_tracker.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/launch/person_tracker.launch -------------------------------------------------------------------------------- /launch/person_tracker_srv.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/launch/person_tracker_srv.launch -------------------------------------------------------------------------------- /package.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/package.xml -------------------------------------------------------------------------------- /resources/example.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/resources/example.gif -------------------------------------------------------------------------------- /resources/tracking.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/resources/tracking.gif -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/deep_sort/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/README.md -------------------------------------------------------------------------------- /src/deep_sort/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/__init__.py -------------------------------------------------------------------------------- /src/deep_sort/deep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/deep_sort/deep/aligned_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/aligned_feature_extractor.py -------------------------------------------------------------------------------- /src/deep_sort/deep/checkpoint/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/deep_sort/deep/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/evaluate.py -------------------------------------------------------------------------------- /src/deep_sort/deep/face_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/face_feature_extractor.py -------------------------------------------------------------------------------- /src/deep_sort/deep/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/feature_extractor.py -------------------------------------------------------------------------------- /src/deep_sort/deep/inception_resnet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/inception_resnet_v1.py -------------------------------------------------------------------------------- /src/deep_sort/deep/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/model.py -------------------------------------------------------------------------------- /src/deep_sort/deep/osnet_ain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/osnet_ain.py -------------------------------------------------------------------------------- /src/deep_sort/deep/reid_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep/reid_feature_extractor.py -------------------------------------------------------------------------------- /src/deep_sort/deep_sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/deep_sort.py -------------------------------------------------------------------------------- /src/deep_sort/sort/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/deep_sort/sort/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/detection.py -------------------------------------------------------------------------------- /src/deep_sort/sort/iou_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/iou_matching.py -------------------------------------------------------------------------------- /src/deep_sort/sort/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/kalman_filter.py -------------------------------------------------------------------------------- /src/deep_sort/sort/linear_assignment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/linear_assignment.py -------------------------------------------------------------------------------- /src/deep_sort/sort/nn_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/nn_matching.py -------------------------------------------------------------------------------- /src/deep_sort/sort/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/preprocessing.py -------------------------------------------------------------------------------- /src/deep_sort/sort/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/track.py -------------------------------------------------------------------------------- /src/deep_sort/sort/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/deep_sort/sort/tracker.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/README.md -------------------------------------------------------------------------------- /src/detector/YOLOv3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/__init__.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/coco.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/coco.data -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/coco.names: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/coco.names -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/darknet19_448.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/darknet19_448.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/tiny-yolo-voc.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/tiny-yolo-voc.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/tiny-yolo.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/tiny-yolo.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/voc.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/voc.data -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/voc.names: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/voc.names -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/voc_gaotie.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/voc_gaotie.data -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/yolo-voc.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/yolo-voc.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/yolo.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/yolo.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/yolo_v3.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/yolo_v3.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/cfg/yolov3-tiny.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/cfg/yolov3-tiny.cfg -------------------------------------------------------------------------------- /src/detector/YOLOv3/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/darknet.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/demo/004545.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/demo/004545.jpg -------------------------------------------------------------------------------- /src/detector/YOLOv3/demo/results/004545.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/demo/results/004545.jpg -------------------------------------------------------------------------------- /src/detector/YOLOv3/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/detect.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/detector.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/__init__.py: -------------------------------------------------------------------------------- 1 | from .nms import boxes_nms -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/build.sh -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/build.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/cpu/nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/cpu/nms_cpu.cpp -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/cpu/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/cpu/vision.h -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/cuda/nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/cuda/nms.cu -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/cuda/vision.h -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/nms.h -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/ext/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/ext/vision.cpp -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/nms.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/nms/python_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/nms/python_nms.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/region_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/region_layer.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/weight/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/detector/YOLOv3/yolo_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/yolo_layer.py -------------------------------------------------------------------------------- /src/detector/YOLOv3/yolo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/YOLOv3/yolo_utils.py -------------------------------------------------------------------------------- /src/detector/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/detector/__init__.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/configs/deep_sort.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/configs/deep_sort.yaml -------------------------------------------------------------------------------- /src/utils/configs/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/configs/yolov3.yaml -------------------------------------------------------------------------------- /src/utils/configs/yolov3_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/configs/yolov3_tiny.yaml -------------------------------------------------------------------------------- /src/utils/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/draw.py -------------------------------------------------------------------------------- /src/utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/evaluation.py -------------------------------------------------------------------------------- /src/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/io.py -------------------------------------------------------------------------------- /src/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/log.py -------------------------------------------------------------------------------- /src/utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/src/utils/parser.py -------------------------------------------------------------------------------- /src/weights/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /srv/choose_target.srv: -------------------------------------------------------------------------------- 1 | int32 target 2 | --- 3 | bool success -------------------------------------------------------------------------------- /srv/clear_target.srv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/srv/clear_target.srv -------------------------------------------------------------------------------- /srv/update_tracker.srv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/khayliang/person_tracking_ros/HEAD/srv/update_tracker.srv --------------------------------------------------------------------------------