From 6d7c417439a9ce72e73d7957ac0f92969aa88b5a Mon Sep 17 00:00:00 2001 From: Trevor Golden Date: Mon, 16 Oct 2023 16:18:19 -0800 Subject: [PATCH] Add catch for exceptions coming out of IfcbBinDataset --- neuston_net.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/neuston_net.py b/neuston_net.py index be1e81a..8499cc2 100755 --- a/neuston_net.py +++ b/neuston_net.py @@ -250,7 +250,12 @@ def do_run(args): print('{} result-file(s) already exist - skipping this bin'.format(bin_obj)) continue - bin_dataset = IfcbBinDataset(bin_fileset, classifier.hparams.resize, classifier.hparams.img_norm) + try: + bin_dataset = IfcbBinDataset(bin_fileset, classifier.hparams.resize, classifier.hparams.img_norm) + except Exception as e: + error_bins.append((bin_obj, e)) + continue + image_loader = DataLoader(bin_dataset, batch_size=args.batch_size, pin_memory=True, num_workers=args.loaders)