├── .gitignore ├── README.md ├── common ├── __init__.py ├── bdd_dataset.py ├── coco_dataset.py ├── data_transforms.py ├── loss_curve.png └── utils.py ├── data ├── demo ├── detect.py ├── params.py ├── predictions.jpg └── predictions_small.jpg ├── evaluate ├── coco_index2category.json ├── eval.py ├── eval_coco.py └── params.py ├── nets ├── __init__.py ├── backbone │ ├── __init__.py │ └── darknet.py ├── model_main.py └── yolo_loss.py ├── requirements.txt ├── test ├── params.py ├── test_fps.py └── test_img.py ├── training ├── params.py ├── training.py └── training_gray.py └── weights └── README.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/README.md -------------------------------------------------------------------------------- /common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /common/bdd_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/common/bdd_dataset.py -------------------------------------------------------------------------------- /common/coco_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/common/coco_dataset.py -------------------------------------------------------------------------------- /common/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/common/data_transforms.py -------------------------------------------------------------------------------- /common/loss_curve.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/common/loss_curve.png -------------------------------------------------------------------------------- /common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/common/utils.py -------------------------------------------------------------------------------- /data: -------------------------------------------------------------------------------- 1 | /media/hd/Dataset/pytorch_data/data -------------------------------------------------------------------------------- /demo/detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/demo/detect.py -------------------------------------------------------------------------------- /demo/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/demo/params.py -------------------------------------------------------------------------------- /demo/predictions.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/demo/predictions.jpg -------------------------------------------------------------------------------- /demo/predictions_small.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/demo/predictions_small.jpg -------------------------------------------------------------------------------- /evaluate/coco_index2category.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/evaluate/coco_index2category.json -------------------------------------------------------------------------------- /evaluate/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/evaluate/eval.py -------------------------------------------------------------------------------- /evaluate/eval_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/evaluate/eval_coco.py -------------------------------------------------------------------------------- /evaluate/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/evaluate/params.py -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nets/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/nets/backbone/__init__.py -------------------------------------------------------------------------------- /nets/backbone/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/nets/backbone/darknet.py -------------------------------------------------------------------------------- /nets/model_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/nets/model_main.py -------------------------------------------------------------------------------- /nets/yolo_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/nets/yolo_loss.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/test/params.py -------------------------------------------------------------------------------- /test/test_fps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/test/test_fps.py -------------------------------------------------------------------------------- /test/test_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/test/test_img.py -------------------------------------------------------------------------------- /training/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/training/params.py -------------------------------------------------------------------------------- /training/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/training/training.py -------------------------------------------------------------------------------- /training/training_gray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/training/training_gray.py -------------------------------------------------------------------------------- /weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhanghanduo/yolo3_pytorch/HEAD/weights/README.md --------------------------------------------------------------------------------