![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 85
杭州的平湖秋月
这个作者很懒,什么都没留下…
展开
-
使用 TensorRT 对 YOLOv8 模型进行 int8 量化
对 YOLOv8 模型进行 int8 量化,主要用到 Python 版的 TensorRT。原创 2023-09-27 23:54:39 · 4271 阅读 · 72 评论 -
用 Vision Transformer 做物体探测任务 object detection
Transformer 最开始是应用于 NLP 的翻译任务,而后续的实践则证明了 Vision Transformer 也能够用于物体探测等 CV 任务。下面是用 Vision Transformer 实现了一个不使用卷积、无预设框 anchor free 的物体探测器,用该探测器在 COCO 数据集上做探测任务的效果。训练环境配置: Keras/TensorFlow 2.9,Python 3.10,WIN 10,Anaconda,Pycharm 以及 Jupyter Lab.原创 2022-09-17 22:00:44 · 2440 阅读 · 1 评论 -
TensorFlow 2.4, Cuda11, Ubuntu 20.04 和 3090 显卡安装
2022 年 5 月更新 Win10 相关内容:经过 1 年多的实际使用,认为还是 Anaconda 最为方便,并且便于在 Win10 环境下使用。使用 Anaconda 的主要好处在于,能够轻松安装多个虚拟环境,比如 TensorFlow 2.9,TensorFlow 2.8 等等。使用时可以根据需要,自由地切换不同版本。例如在 Win10 中,之前使用 TF 2.8,现在要在 Anaconda 中新装一个 TF 2.9 的虚拟环境,运行 conda create -n tf2.9 python=3原创 2021-01-10 20:43:28 · 1316 阅读 · 0 评论 -
Keras/TensorFlow 交叉熵损失函数中的 from_logits 参数
文章目录1. from_logits 的两种使用方法。2. from_logits=True 是一个简便操作。3. logit 与 sigmoid 互为反函数。1. from_logits 的两种使用方法。在 Keras 的二元交叉熵损失函数(binary crossentropy,下面简称为 bce)中,有 from_logits 这个参数。在使用时需要分下面 2 种情况:如果输入给 bce 的是一个 logit 值(大小范围 [-∞, +∞] ),则应该设置 from_logits=Tru原创 2022-03-03 11:05:16 · 5722 阅读 · 4 评论 -
使用混合精度 mixed precision 给 Keras 加速
可以使用混合精度 mixed precision 给 Keras 加速,3个操作步骤如下:使用算力在 7.0以上的GPU,比如 NVIDIA的 RTX 3090, 3080等。在建立模型之前,设置 global_policy 为 mixed_float16。policy = tf.keras.mixed_precision.Policy('mixed_float16')tf.keras.mixed_precision.set_global_policy(policy)# 另一种写法是只用下面这原创 2021-11-16 22:58:29 · 2222 阅读 · 0 评论 -
用 Keras/TensorFlow 2.9 创建深度学习模型的方法总结
前言Keras 中有 3 种方法可以创建深度学习模型:1. keras.Sequential 模型。 2.函数式 API。3.子类法 subclassing keras.Model。因为可用的方法很多,灵活性大,实际使用时容易出错。但是如果建模的目标,是能够表达任意结构的深度学习模型,并且能够看到模型的完整内部结构,那么在创建模型时,有 3 个简单的原则可以使用。.........原创 2022-06-04 19:30:13 · 4957 阅读 · 2 评论