MMpretrain
文章平均质量分 93
qq_41627642
这个作者很懒,什么都没留下…
展开
-
开源预训练框架 MMPRETRAIN现有的推理模型(三)
(1)ImageClassificationInferencer:对给定图像进行图像分类。(2)ImageRetrievalInferencer:从给定图像集上的给定图像执行图像到图像检索。(3)ImageCaptionInferencer:在给定图像上生成标题(4)VisualQuestionAnsweringInferencer:根据给定的图片回答问题。(5)VisualGroundingInferencer:从给定图像的描述中找到一个对象。原创 2023-07-27 18:05:30 · 571 阅读 · 0 评论 -
开源预训练框架 MMPRETRAIN官方文档(高级指南)
在新的数据集教程中,我们知道数据集类使用该load_data_list方法来初始化整个数据集,并将每个样本的信息保存到一个字典中。通常,为了节省内存使用,我们只加载load_data_list 中的图像路径和标签,并在使用时加载完整的图像内容。此外,我们可能希望在训练时选取样本时进行一些随机数据增强。几乎所有的数据加载、预处理和格式化操作都可以通过数据管道在 MMPretrain 中配置。数据管道意味着当从数据集中索引样本时如何处理样本字典。它由一系列数据转换组成。原创 2023-07-13 16:45:10 · 1991 阅读 · 1 评论 -
开源预训练框架 MMPRETRAIN官方文档(概览、环境安装与验证、基础用户指南)
在本节中,我们将演示如何使用 PyTorch 准备环境。MMPretrain 适用于 Linux、Windows 和 macOS。它需要 Python 3.7+、CUDA 10.2+ 和 PyTorch 1.8+。在本教程中,我们提供了一个练习示例以及一些有关如何在您自己的数据集上进行训练的技巧。在 MMPretrain 中,我们支持CustomDataset(类似于ImageFolderin torchvision),可以直接读取指定文件夹内的图像。原创 2023-07-07 17:53:01 · 5317 阅读 · 1 评论