├── README.md ├── __pycache__ ├── models.cpython-36.pyc └── test.cpython-36.pyc ├── assets ├── dog.png ├── giraffe.png ├── messi.png └── traffic.png ├── check.py ├── config ├── coco.data ├── create_custom_model.sh ├── custom.data ├── yolov3-tiny.cfg └── yolov3.cfg ├── data ├── 5k.txt ├── coco.names ├── custom │ ├── classes.names │ ├── images │ │ └── train.jpg │ ├── labels │ │ └── train.txt │ ├── train.txt │ └── valid.txt ├── get_coco_dataset.sh ├── samples │ ├── dog.jpg │ ├── eagle.jpg │ ├── field.jpg │ ├── giraffe.jpg │ ├── herd_of_horses.jpg │ ├── messi.jpg │ ├── person.jpg │ ├── room.jpg │ └── street.jpg └── temptest │ └── field.jpg ├── detect.py ├── models.py ├── output ├── dog.png ├── eagle.png ├── field.png ├── giraffe.png ├── herd_of_horses.png ├── messi.png ├── person.png ├── room.png └── street.png ├── requirements.txt ├── test.py ├── train.py ├── utils ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-36.pyc │ ├── augmentations.cpython-36.pyc │ ├── datasets.cpython-36.pyc │ ├── logger.cpython-36.pyc │ ├── parse_config.cpython-36.pyc │ └── utils.cpython-36.pyc ├── augmentations.py ├── datasets.py ├── logger.py ├── parse_config.py └── utils.py └── weights └── download_weights.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/README.md -------------------------------------------------------------------------------- /__pycache__/models.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/__pycache__/models.cpython-36.pyc -------------------------------------------------------------------------------- /__pycache__/test.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/__pycache__/test.cpython-36.pyc -------------------------------------------------------------------------------- /assets/dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/assets/dog.png -------------------------------------------------------------------------------- /assets/giraffe.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/assets/giraffe.png -------------------------------------------------------------------------------- /assets/messi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/assets/messi.png -------------------------------------------------------------------------------- /assets/traffic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/assets/traffic.png -------------------------------------------------------------------------------- /check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/check.py -------------------------------------------------------------------------------- /config/coco.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/config/coco.data -------------------------------------------------------------------------------- /config/create_custom_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/config/create_custom_model.sh -------------------------------------------------------------------------------- /config/custom.data: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/config/custom.data -------------------------------------------------------------------------------- /config/yolov3-tiny.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/config/yolov3-tiny.cfg -------------------------------------------------------------------------------- /config/yolov3.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/config/yolov3.cfg -------------------------------------------------------------------------------- /data/5k.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/coco.names: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/coco.names -------------------------------------------------------------------------------- /data/custom/classes.names: -------------------------------------------------------------------------------- 1 | train 2 | -------------------------------------------------------------------------------- /data/custom/images/train.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/custom/images/train.jpg -------------------------------------------------------------------------------- /data/custom/labels/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/custom/labels/train.txt -------------------------------------------------------------------------------- /data/custom/train.txt: -------------------------------------------------------------------------------- 1 | data/custom/images/train.jpg 2 | -------------------------------------------------------------------------------- /data/custom/valid.txt: -------------------------------------------------------------------------------- 1 | data/custom/images/train.jpg 2 | -------------------------------------------------------------------------------- /data/get_coco_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/get_coco_dataset.sh -------------------------------------------------------------------------------- /data/samples/dog.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/dog.jpg -------------------------------------------------------------------------------- /data/samples/eagle.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/eagle.jpg -------------------------------------------------------------------------------- /data/samples/field.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/field.jpg -------------------------------------------------------------------------------- /data/samples/giraffe.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/giraffe.jpg -------------------------------------------------------------------------------- /data/samples/herd_of_horses.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/herd_of_horses.jpg -------------------------------------------------------------------------------- /data/samples/messi.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/messi.jpg -------------------------------------------------------------------------------- /data/samples/person.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/person.jpg -------------------------------------------------------------------------------- /data/samples/room.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/room.jpg -------------------------------------------------------------------------------- /data/samples/street.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/samples/street.jpg -------------------------------------------------------------------------------- /data/temptest/field.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/data/temptest/field.jpg -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/detect.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/models.py -------------------------------------------------------------------------------- /output/dog.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/dog.png -------------------------------------------------------------------------------- /output/eagle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/eagle.png -------------------------------------------------------------------------------- /output/field.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/field.png -------------------------------------------------------------------------------- /output/giraffe.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/giraffe.png -------------------------------------------------------------------------------- /output/herd_of_horses.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/herd_of_horses.png -------------------------------------------------------------------------------- /output/messi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/messi.png -------------------------------------------------------------------------------- /output/person.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/person.png -------------------------------------------------------------------------------- /output/room.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/room.png -------------------------------------------------------------------------------- /output/street.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/output/street.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/augmentations.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/augmentations.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/datasets.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/datasets.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/logger.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/parse_config.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/parse_config.cpython-36.pyc -------------------------------------------------------------------------------- /utils/__pycache__/utils.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/__pycache__/utils.cpython-36.pyc -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/parse_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/parse_config.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/utils/utils.py -------------------------------------------------------------------------------- /weights/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DocF/YOLOv3-Torch2TRT/HEAD/weights/download_weights.sh --------------------------------------------------------------------------------