Github 项目推荐 | YOLOv3 的最小化 PyTorch 实现

AI研习社

发布于 2018-07-26 10:51:18

1.2K0

发布于 2018-07-26 10:51:18

该库给 YOLOv3 提供了一些更新，并且做了些小的设计使其变得更好，同时还训练了新的网络。它更加精确，运行速度同样很快。在 320×320 YOLOv3 上以 22.2 mAP 运行 22 ms，与 SSD 一样准确，但速度提高了三倍。在 Titan X上，它能在 51 ms 内达到 57.9 AP50，相比较而言，RetinaNet 在 198 ms 内达到 57.5 AP50，性能相似但速度快了 3.8 倍。和往常一样，所有代码都可在 https://pjreddie.com/yolo/ 上找到。

Github：

https://github.com/eriklindernoren/PyTorch-YOLOv3

安装：

$ git clone https://github.com/eriklindernoren/PyTorch-YOLOv3
$ cd PyTorch-YOLOv3/
$ sudo pip3 install -r requirements.txt

下载预训练权值

$ cd weights/
$ bash download_weights.sh

下载 COCO

$ cd data/
$ bash get_coco_dataset.sh

使用预训练权值训练图像，下图显示了将输入图像缩放为 256x256 时的推理时间。

$ python3 detect.py --image_folder /data/samples

测试

在 COCO 测试中评估模型。

$ python3 test.py --weights_path weights/yolov3.weights

训练

在 COCO 上训练，数据增强和其他训练技巧有待优化。

train.py [-h] [--epochs EPOCHS] [--image_folder IMAGE_FOLDER]
                [--batch_size BATCH_SIZE]
                [--model_config_path MODEL_CONFIG_PATH]
                [--data_config_path DATA_CONFIG_PATH]
                [--weights_path WEIGHTS_PATH] [--class_path CLASS_PATH]
                [--conf_thres CONF_THRES] [--nms_thres NMS_THRES]
                [--n_cpu N_CPU] [--img_size IMG_SIZE]
                [--checkpoint_interval CHECKPOINT_INTERVAL]
                [--checkpoint_dir CHECKPOINT_DIR]

Credit

@article{yolov3,
  title={YOLOv3: An Incremental Improvement},
  author={Redmon, Joseph and Farhadi, Ali},
  journal = {arXiv},
  year={2018}

本文参与腾讯云自媒体分享计划，分享自微信公众号。

原始发表：2018-06-01，如有侵权请联系 cloudcommunity@tencent.com 删除

github

pytorch

https

本文分享自 AI研习社微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体分享计划，欢迎热爱写作的你一起参与！

github

pytorch

https

登录后参与评论

0 条评论

热度