├── .gitignore ├── CMakeLists.txt ├── How_to_build_and_run.md ├── README.md ├── launch └── detector.launch ├── msg ├── BoundingBox.msg └── BoundingBoxes.msg ├── package.xml ├── rviz.png ├── view_img.jpg └── yolov5 ├── data ├── argoverse_hd.yaml ├── coco.yaml ├── coco128.yaml ├── hyp.finetune.yaml ├── hyp.scratch.yaml ├── images │ ├── bus.jpg │ └── zidane.jpg ├── scripts │ ├── get_argoverse_hd.sh │ ├── get_coco.sh │ └── get_voc.sh └── voc.yaml ├── detect.py ├── hubconf.py ├── model.py ├── models ├── __init__.py ├── common.py ├── experimental.py ├── export.py ├── hub │ ├── anchors.yaml │ ├── yolov3-spp.yaml │ ├── yolov3-tiny.yaml │ ├── yolov3.yaml │ ├── yolov5-fpn.yaml │ ├── yolov5-p2.yaml │ ├── yolov5-p6.yaml │ ├── yolov5-p7.yaml │ ├── yolov5-panet.yaml │ ├── yolov5l6.yaml │ ├── yolov5m6.yaml │ ├── yolov5s6.yaml │ └── yolov5x6.yaml ├── yolo.py ├── yolov5l.yaml ├── yolov5m.yaml ├── yolov5s.yaml └── yolov5x.yaml ├── requirements.txt ├── utils ├── __init__.py ├── activations.py ├── autoanchor.py ├── aws │ ├── __init__.py │ ├── mime.sh │ ├── resume.py │ └── userdata.sh ├── datasets.py ├── general.py ├── google_app_engine │ ├── Dockerfile │ ├── additional_requirements.txt │ └── app.yaml ├── google_utils.py ├── loss.py ├── metrics.py ├── parse_config.py ├── plots.py ├── torch_utils.py ├── utils.py └── wandb_logging │ ├── __init__.py │ ├── log_dataset.py │ └── wandb_utils.py └── weights ├── download_weights.sh └── yolov5s.pt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /How_to_build_and_run.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/How_to_build_and_run.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/README.md -------------------------------------------------------------------------------- /launch/detector.launch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/launch/detector.launch -------------------------------------------------------------------------------- /msg/BoundingBox.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/msg/BoundingBox.msg -------------------------------------------------------------------------------- /msg/BoundingBoxes.msg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/msg/BoundingBoxes.msg -------------------------------------------------------------------------------- /package.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/package.xml -------------------------------------------------------------------------------- /rviz.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/rviz.png -------------------------------------------------------------------------------- /view_img.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/view_img.jpg -------------------------------------------------------------------------------- /yolov5/data/argoverse_hd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/argoverse_hd.yaml -------------------------------------------------------------------------------- /yolov5/data/coco.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/coco.yaml -------------------------------------------------------------------------------- /yolov5/data/coco128.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/coco128.yaml -------------------------------------------------------------------------------- /yolov5/data/hyp.finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/hyp.finetune.yaml -------------------------------------------------------------------------------- /yolov5/data/hyp.scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/hyp.scratch.yaml -------------------------------------------------------------------------------- /yolov5/data/images/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/images/bus.jpg -------------------------------------------------------------------------------- /yolov5/data/images/zidane.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/images/zidane.jpg -------------------------------------------------------------------------------- /yolov5/data/scripts/get_argoverse_hd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/scripts/get_argoverse_hd.sh -------------------------------------------------------------------------------- /yolov5/data/scripts/get_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/scripts/get_coco.sh -------------------------------------------------------------------------------- /yolov5/data/scripts/get_voc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/scripts/get_voc.sh -------------------------------------------------------------------------------- /yolov5/data/voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/data/voc.yaml -------------------------------------------------------------------------------- /yolov5/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/detect.py -------------------------------------------------------------------------------- /yolov5/hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/hubconf.py -------------------------------------------------------------------------------- /yolov5/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/model.py -------------------------------------------------------------------------------- /yolov5/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov5/models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/common.py -------------------------------------------------------------------------------- /yolov5/models/experimental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/experimental.py -------------------------------------------------------------------------------- /yolov5/models/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/export.py -------------------------------------------------------------------------------- /yolov5/models/hub/anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/anchors.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov3-spp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov3-spp.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov3-tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov3-tiny.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov3.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5-fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5-fpn.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5-p2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5-p2.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5-p6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5-p6.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5-p7.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5-p7.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5-panet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5-panet.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5l6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5l6.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5m6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5m6.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5s6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5s6.yaml -------------------------------------------------------------------------------- /yolov5/models/hub/yolov5x6.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/hub/yolov5x6.yaml -------------------------------------------------------------------------------- /yolov5/models/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/yolo.py -------------------------------------------------------------------------------- /yolov5/models/yolov5l.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/yolov5l.yaml -------------------------------------------------------------------------------- /yolov5/models/yolov5m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/yolov5m.yaml -------------------------------------------------------------------------------- /yolov5/models/yolov5s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/yolov5s.yaml -------------------------------------------------------------------------------- /yolov5/models/yolov5x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/models/yolov5x.yaml -------------------------------------------------------------------------------- /yolov5/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/requirements.txt -------------------------------------------------------------------------------- /yolov5/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov5/utils/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/activations.py -------------------------------------------------------------------------------- /yolov5/utils/autoanchor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/autoanchor.py -------------------------------------------------------------------------------- /yolov5/utils/aws/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov5/utils/aws/mime.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/aws/mime.sh -------------------------------------------------------------------------------- /yolov5/utils/aws/resume.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/aws/resume.py -------------------------------------------------------------------------------- /yolov5/utils/aws/userdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/aws/userdata.sh -------------------------------------------------------------------------------- /yolov5/utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/datasets.py -------------------------------------------------------------------------------- /yolov5/utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/general.py -------------------------------------------------------------------------------- /yolov5/utils/google_app_engine/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/google_app_engine/Dockerfile -------------------------------------------------------------------------------- /yolov5/utils/google_app_engine/additional_requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/google_app_engine/additional_requirements.txt -------------------------------------------------------------------------------- /yolov5/utils/google_app_engine/app.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/google_app_engine/app.yaml -------------------------------------------------------------------------------- /yolov5/utils/google_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/google_utils.py -------------------------------------------------------------------------------- /yolov5/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/loss.py -------------------------------------------------------------------------------- /yolov5/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/metrics.py -------------------------------------------------------------------------------- /yolov5/utils/parse_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/parse_config.py -------------------------------------------------------------------------------- /yolov5/utils/plots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/plots.py -------------------------------------------------------------------------------- /yolov5/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/torch_utils.py -------------------------------------------------------------------------------- /yolov5/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/utils.py -------------------------------------------------------------------------------- /yolov5/utils/wandb_logging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /yolov5/utils/wandb_logging/log_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/wandb_logging/log_dataset.py -------------------------------------------------------------------------------- /yolov5/utils/wandb_logging/wandb_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/utils/wandb_logging/wandb_utils.py -------------------------------------------------------------------------------- /yolov5/weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/weights/download_weights.sh -------------------------------------------------------------------------------- /yolov5/weights/yolov5s.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Shua-Kang/ros_pytorch_yolov5/HEAD/yolov5/weights/yolov5s.pt --------------------------------------------------------------------------------