pix2pixHD代码---readme

news/2024/12/1 19:53:23/

1:基础配置
要求大于等于11G的显卡,安装pytorch,下载代码。
在这里插入图片描述
2:测试
dataset文件中放的是一些例子,下载cityscape的预训练权重,放入到checkpoints文件夹下,测试模型。测试结果放在results文件夹下的label2city_1024p文件夹下。
在这里插入图片描述
3:数据集
首先下载Cityscape数据集,接着放置在datasets文件夹下,要和example放置的一样
在这里插入图片描述
example放置:里面每个文件下的图片都是一一对应的。
在这里插入图片描述
下载后的原始数据集:
在这里插入图片描述
每一个文件夹下都有三个文件夹,里面放的是train,test,val。
在这里插入图片描述
每一个train,test,val下放的都是城市的名字,每个城市名字下放的是图片。
在这里插入图片描述
4:训练
训练使用512x1024大小的数据集。
在这里插入图片描述
5:多GPU训练
在train的时候制定GPU的型号即可。
在这里插入图片描述
6:混合精度
训练时候制定使用混合精度即可,前提是安装了APEX。
在这里插入图片描述
7:如果要在训练时候使用全分辨率,需要24G以上的内存,否则就裁切到512x1024.在这里插入图片描述
8:训练自己的数据集
8.1:首先生成一个通道的寓意标签,标签上的每个值是物体对应的类别数,因为我们要从标签图中生成one-hot编码。在训练和测试时候制定通道数(类别数,cityscape未转换前是35个类别)。
8.2:如果输入不是标签图的话,指定label_nc=0,这将会直接使用RGB图片,dataset的名称也应该由train_label,train_img替换为train_A,train_B。即将A转换为B。
8.3:如果没有实例图,指定no_instance。
8.4:默认的预处理方式是scale_with,首先会将训练图片的宽缩放到1024,同时保持横纵比,长就会变为512(横纵比2:1),如果想用不同的设置,通过使用resize_or_crop改变他。比如,scale width and crop首先将图片的宽缩放到1024,然后随机crop到(finesize,finesize),长和宽一样。crop的话跳过resize操作,随机剪切。不想预处理的话就设置为None。
在这里插入图片描述
9:more
在这里插入图片描述


http://www.ppmy.cn/news/79688.html

相关文章

电力系统的虚假数据注入攻击和MTD系统研究(Matlab代码实现)

💥💥💞💞欢迎来到本博客❤️❤️💥💥 🏆博主优势:🌞🌞🌞博客内容尽量做到思维缜密,逻辑清晰,为了方便读者。 ⛳️座右铭&a…

AI孙燕姿爆火,华语乐坛重温旧梦

最近在B站,AI孙燕姿火了。 浏览相关搜索页面,这位新晋“红人”翻唱了各种类型的歌曲,包括《红豆》《爱在西元前》《水星记》《安河桥》,甚至还有《向天再借五百年》,更重要的是,表现普遍不俗,俘…

BERT网络的原理与实战

BERT网络的原理与实战 一、简介二、原理1. Transformer2. BERT2.1 MLM2.2 NSP 3. Fine-tuning 三、实战1. 数据集2. 预处理3. 模型训练 一、简介 BERT(Bidirectional Encoder Representations from Transformers)是一种基于Transformer架构的预训练语言…

前端搭建砸地鼠游戏(内附源码)

The sand accumulates to form a pagoda ✨ 写在前面✨ 功能介绍✨ 页面搭建✨ 样式设置✨ 逻辑部分✨ 完整代码 ✨ 写在前面 上周我们实通过前端基础实现了打字通,当然很多伙伴再评论区提出了想法,后续我们会考虑实现的,今天还是继续按照我…

kafka基础介绍

目录 前言: 一:kafka架构 1.kafka基础架构 2、kafka多副本架构 二、kafka基础概念 1、produce 2. Consumer 3、Broker ​ 4、Topic 5、Partition 6、Replicas 7、Offset 8、 AR 9、 ISR 10、OSR 11、HW 12、LEO 13、Lag 三、kafka特性 四、kafka…

YOLO中的值得借鉴的思想

关键理论的理解,后面会补充结构等。 1.YOLO1中将图像划分为7*7个网格,每个网格都预测网格中的的类别(是什么物体),以及预测到的物体所对应的框(四个位置量,一个置信度),所…

[golang gin框架] 37.ElasticSearch 全文搜索引擎的使用

一.全文搜索引擎 ElasticSearch 的介绍,以 及安装配置前的准备工作 介绍 ElasticSearch 是一个基于 Lucene 的 搜索服务器,它提供了一个 分布式多用户能力的 全文搜索引擎,基于 RESTful web 接口,Elasticsearch 是用 Java 开发的,并作为 Apac…

MATLAB|004|MATLAB M-Files|MATLAB数据类型

MATLAB M-Files 我们一直强调MATLAB是一个功能强大的编程语言及交互式计算环境,之前学习的内容中,我们学会了在 MATLAB 命令提示符下输入命令,而且我们主要是把MATLAB环境作为一个计算器使用。其实,MATLAB 还允许写入到一个文件中…