Graduation-Project/federated_learning/GenerateTestdata.sh

16 lines
654 B
Bash
Raw Permalink Normal View History

2025-04-20 07:20:40 +00:00
# 创建测试目录结构
mkdir -p ./test_data/{client1,client2}/{train,val}/images
mkdir -p ./test_data/{client1,client2}/{train,val}/labels
# 生成虚拟数据各客户端仅需2张图片
for client in client1 client2; do
for split in train val; do
# 创建空图片128x128 RGB
magick -size 128x128 xc:white test_data/${client}/${split}/images/img1.jpg
magick -size 128x128 xc:black test_data/${client}/${split}/images/img2.jpg
# 创建示例标签文件
echo "0 0.5 0.5 0.2 0.2" > test_data/${client}/${split}/labels/img1.txt
echo "1 0.3 0.3 0.4 0.4" > test_data/${client}/${split}/labels/img2.txt
done
done