mirror of
https://github.com/unanmed/ginka-generator.git
synced 2026-05-14 04:41:12 +08:00
17 lines
722 B
Python
17 lines
722 B
Python
import torch
|
|
|
|
def random_smooth_onehot(onehot_map, min_main=0.75, max_main=1.0, epsilon=0.25):
|
|
"""
|
|
生成随机平滑的 one-hot 编码，使主类别概率不再固定，而是随机波动
|
|
"""
|
|
C, H, W = onehot_map.shape
|
|
# 生成主类别的随机概率 (min_main, max_main)
|
|
main_prob = torch.rand(H, W) * (max_main - min_main) + min_main
|
|
|
|
# 计算剩余概率并随机分配到其他类别
|
|
noise = torch.rand(C, H, W) * epsilon # 随机噪声
|
|
noise = noise / noise.sum(dim=1, keepdim=True) # 归一化到总和为 epsilon
|
|
|
|
# 计算最终平滑 one-hot 结果
|
|
smooth_onehot = onehot_map * main_prob + (1 - onehot_map) * noise
|
|
return smooth_onehot |