| 1234567891011121314151617181920 |
- # For 1) 用于数据处理部分，对于新数据集如何生成对应
- # -------------------------------------------------------------------------------------------------------------------- #
- python ./tool/generate_txt.py --txt_path './dataset/New_dataset' --specific_data 'testtest' --txt_name 'train'
- # 以CIFAR-10 数据为例，需要生成train和test的txt文件应该是这样的：
- python ./tool/generate_txt.py --txt_path './dataset/CIFAR-10_ori' --specific_data 'train_cifar10_JPG' --txt_name 'train'
- python ./tool/generate_txt.py --txt_path './dataset/CIFAR-10_ori' --specific_data 'test_cifar10_JPG' --txt_name 'test'
- # For 2)用于水印植入处理部分 字符串插入图像处理
- # -------------------------------------------------------------------------------------------------------------------- #
- # 密钥生成部分文件放置在 './dataset/watermarking'里，其中同时含有key_hex.txt和对应根据classes拆分的 QR images，便于选择水印插入方式
- # For 用于剪枝模型后边缘部署
- # -------------------------------------------------------------------------------------------------------------------- #
|
