Please enable JavaScript.
Coggle requires JavaScript to display documents.
get_bi_dataset - Coggle Diagram
get_bi_dataset
bi类
初始化
加载数据集，这个数据集键对应图片名称（第几个视频第几帧）值对应提取到的68个数据点组成的二维列表，把他们任用字典形式加载到内存中
重复复制帧
搞了一个二维仿射变换图片增强序列
gen_one_datapoint
从之前copy了2000次的记录图片path列表中随机选择一个路径作为背景图路径
随机选择真假
如果是真
生成伪造图和遮罩
调用类里的get_blended_face
找到之前加载到内存里该图片对应的68个数据点
依据这68个数据点到数据集中找最相似的图片
获得背景脸对应的视频和帧数
从之前图片数据集里随机选择5000张
这里就能理解，原来初始化的地方要修改
过滤掉所有来自同一个视频的
通过计算欧式距离寻找最相近的脸
压缩图片，同时也要等比缩小68个数据点
压缩前后脸
获得面罩
先通过4种随机方法获得初始面罩
对初始面罩仿射变化和随机腐蚀膨胀
颜色风格迁移
建立副本
获取 mask 中非零值的索引，以确定需要进行颜色迁移的区域
分别获取原图像和目标图像中与mask匹配的区域
分别计算源和目标图像的平均值
限制颜色区域
存回并返回副本
融合前脸，内脸，面罩
形状还原
加载背景脸图片
用公式2得到遮罩的Xray
如果是假
图片不处理，生成空遮罩
为了增加图片多样性，随机进行下面步骤
图像压缩
下采样
翻折