加载预训练模型
network_pkl = path_to_pretrained_model.pkl
device = torch.device('cuda')
with dnnlib.util.open_url(network_pkl) as f:
G = legacy.load_network_pkl(f)['G_ema'].to(device)
生成图片
z = torch.randn([1, G.z_dim], device=device)
img = G(z, None)
4、数据集增强
如果目的是生成大量相似图片用于数据集扩充,可以使用数据增强技术:
工具:Albumentations、Augmentor、TensorFlow Data Augmentation
方法:对原始图片进行旋转、缩放、裁剪、颜色调整等操作。
示例代码:
python
from tensorflow.keras.preprocessing.image import ImageDataGenerator
import matplotlib.pyplot as plt