在前面计算完损失后，该进行更新：
1：netEMA是模型的生成器：
遍历生成器的state_dict，将每一个键对应的值乘以EMA_decay。
接着根据当前迭代步数计算num_upd，每1000,2500,10000代倍数就执行一次。
当num_upd大于50就跳出更新EMA。
接着对图片进行上色：
def visualize_batch(self, model, image, label, cur_iter):
self.save_images(label, "label", cur_iter, is_label=True)
self.save_images(image, "real", cur_iter)
with torch.no_grad():
model.eval()
fake = model.netG(label)
self.save_images(fake, "fake", cur_iter)
model.train()
if not self.opt.no_EMA:
model.eval()
fake = model.netEMA(label)
self.save_images(fake, "fake_ema", cur_iter)
model.train()
首先对标签进行上色：
label即batch是经过one-hot编码后的标签大小为(5,35,256,512)。
接着len(batch）=5，取第一个batch对应的tensor。
进行上色：
首先获得camp：
生成的colormap包含空像素和噪声，一共有36个类别，所以执行else语句。
首先生成一个全为0的列表，大小为(36,3)—>接着遍历每个类别，初始化r=g=b=0—>id = 1—>接着遍历7次，首先将id转换为二进制类型。
def uint82bin(n, count=8):
"""returns the binary of integer n, count refers to amount of bits"""
return ''.join([str((n >> y) & 1) for y in range(count - 1, -1, -1)])
#y = 7,6,5,4,3,2,1,0
y分别取值为7,6,5,4,3,2,1,0。
将n右移位7位，n为1，则移位后为0，分别移位，只有当y等于0时，不移位，n才为1.最后返回一个字符串’00000001’.
移位操作
分别取str_id的倒数1,2，3位。然后将1,0,0分别左移七位，1左移后变为二进制为1000 0000即128.0左移后还是0，所以r=128,g=b=0.
最后id=1右移3位，变为0.
在j循环里执行8次，则下一次id=0.在uint82bin函数中，0不管位移多少次都为0，且0&1=0，所以最后输出’00000000’.
则r = 128^(0)=128.
因为128=(10000000),0=(00000000),(1异或0=1),(0异或0=0)，所以128^(0)=128。这样执行7次后，将r填充为第一行第一列，g填充为第一行第二列，b填充为第一行第三列。这样执行for循环36次，则camp就会被重新填充一遍。
将camp转换为tensor。生成一个由0填充的(3,256,512)大小的size。同时对label的其中一个batch数据求类别。
tens大小由(35,256,512)变为(1,256,512)。
len(camp)=36，开始label=0时，tens[0]=(256,512)，label==tens[0]会得到一个mask,其中tens中等于0的类别为True，不等于0的为false。
color_image[0]取得color_image第一层R通道，cmap[label][0]为第一行第一列即128，将mask对应的值全部替换为128.同理G和B通道也是这样处理。这样循环36次，将每一个类别都上色。最后输出经过填充的彩色图。
最后将label进行转置，方便cv2保存。
最后将batch剩余的四个图片也进行处理。将五张图放在一个图片上保存到指定位置。
下一步对image处理：
将tens小于0的设置为0，大于1的设置为1.再转置为(h,w,c)格式。
在eval时候，将label输入到生成器中，生成fake image，大小为(5,3,256,512)。将生成的fake image保存起来。
netEMA是对生成器的深拷贝。
下一步计算训练一个batch所需要的时间：
将epoch，总epoch,当前迭代，所花费的时间写到progress.txt文件中，并打印出来。
下一步：
通过控制latest,best来保存权重。
最重要看一下FID计算：比较麻烦，到时候重新开一章。