今日AI:YOLOv5多种backbone、模型压缩代码库开源;FastAI 发布 PyTorch 学习免费书...

点击下方卡片,关注 AIZOO 公众号

AI 重磅干货,第一时间送达

今天,分享一个新发布的基于 YOLOv5 的开源库,提供了多种 backbone和压缩工具,以及 FastAI 发布的一本电子书。

1. YOLOv5-Multibackbone-Compression 多 backbone以及压缩库

4ed47d51653d92d762f0acd0e7538f26.png

e1bfa27d42ca4efb70abc77a97e7fca8.png

代码链接:

https://github.com/Gumpest/YOLOv5-Multibackbone-Compression

目前该仓库有 236 star,提供了多种不同的 backbone,包括以下backbone:

  • TPH-YOLOv5:加 Transformer 的 YOLOv5

  • SwinTrans-YOLOv5:SwinTrans 作为 backbone

  • YOLOv5l-GhostNet

  • YOLOv5l-ShuffleNetV2

  • YOLOv5l-MobileNetv3Small

  • YOLOv5l-EfficientNetLite0

  • YOLOv5l-PP-LCNet

各种 backbone 的 YOLOv5 在 Visdrone 数据集的精度如下:

(1 size is 608,2-6 size is 640)

No Model mAP mAP@50 Parameters(M) GFLOPs TrainCost(h) Memory Cost(G) PT File FPS@CPU
1 YOLOv5l 23.2 39.7 46.19 108.1



2 YOLOv5l-GhostNet 18.4 33.8 24.27 42.4 27.44 4.97 PekingUni Cloud
3 YOLOv5l-ShuffleNetV2 16.48 31.1 21.27 40.5 10.98 2.41 PekingUni Cloud
4 YOLOv5l-MobileNetv3Small 16.55 31.2 20.38 38.4 10.19 5.30 PekingUni Cloud
5 YOLOv5l-EfficientNetLite0 19.12 35 23.01 43.9 13.94 2.04 PekingUni Cloud
6 YOLOv5l-PP-LCNet 17.63 32.8 21.64 41.7 18.52 1.66 PekingUni Cloud

另外,该仓库还提供了各种减枝和量化、训练和评估工具,大大的利好学生啊。

今日AI:YOLOv5多种backbone、模型压缩代码库开源;FastAI 发布 PyTorch 学习免费书..._第1张图片

2. FastAI 发布深度学习电子书

FastAI 是一个深度学习组织,发布了深度深度学习课程和开源代码,还发布了一个基于 PyTorch 的上层库 Fastai。书籍对应代码:https://github.com/fastai/fastbook

书籍下载地址:公众号 AIZOO 后台回复fastai 就可以下载。

书籍封面

今日AI:YOLOv5多种backbone、模型压缩代码库开源;FastAI 发布 PyTorch 学习免费书..._第2张图片

书籍内容包括深度学习介绍、如何训练、深度学习在 CV和NLP的应用,以及如何部署,内容看着挺丰富的,大家可以自行下载和查看。

c6cce6aadd5d2b26ee806135d2af4000.png

欢迎扫描下方的二维码添加小助手微信,邀请您加入我们的微信交流群。

群里有多位清北复交、BAT、AI独角兽大牛和众多深度学习er在一起愉快的交流技术,有任何问题,都可以咨询大家,欢迎你的加入哦。

今日AI:YOLOv5多种backbone、模型压缩代码库开源;FastAI 发布 PyTorch 学习免费书..._第3张图片

 添加小助手微信,邀您进 AIZOO 技术交流群

听说点个在看的人运气都很好~

你可能感兴趣的:(人工智能,github,深度学习,神经网络,机器学习)