colab-notebookhigh-resolutioncolabgenerative-adversarial-networkscnngenerative-adversarial-networkganfourier-transformfourier-convolutionspytorchfourierinpainting-methodsdeep-neural-networksinpainting-algorithmdeep-learninginpaintingcomputer-visionimage-inpainting
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
470 B
22 lines
470 B
#!/usr/bin/env python3 |
|
|
|
|
|
import tqdm |
|
import webdataset as wds |
|
|
|
|
|
def main(args): |
|
input_dataset = wds.Dataset(args.infile) |
|
output_dataset = wds.ShardWriter(args.outpattern) |
|
for rec in tqdm.tqdm(input_dataset): |
|
output_dataset.write(rec) |
|
|
|
|
|
if __name__ == '__main__': |
|
import argparse |
|
|
|
aparser = argparse.ArgumentParser() |
|
aparser.add_argument('infile', type=str) |
|
aparser.add_argument('outpattern', type=str) |
|
|
|
main(aparser.parse_args())
|
|
|