12345678910111213141516171819202122232425 |
- from watermark_generate.tools.image_classify_dataset_process import deal_img_label
- if __name__ == '__main__':
-
- dataset_dir = './dataset/imagenette2-320/train'
- dst_img_dir = './dataset/VOC2007_QR/JPEGImages'
- trigger_dataset_dir = './dataset/trigger'
- trigger_upload_dir = '../watermark_generate/extracted/'
- watermark_gen_dir = './dataset/watermarking'
- bbox_filename = './dataset/qrcode_positions.txt'
- deal_img_label(watermarking_dir=watermark_gen_dir, dataset_dir=dataset_dir, num_samples=2, dst_img_dir=None,
- prefix='wm', trigger=False, bbox_filename=bbox_filename)
- dataset_dir = './dataset/imagenette2-320/val'
- deal_img_label(watermarking_dir=watermark_gen_dir, dataset_dir=dataset_dir, num_samples=2, dst_img_dir=None,
- prefix='wm', trigger=False, bbox_filename=bbox_filename)
-
-
-
-
-
-
-
|