Skip to content

Instantly share code, notes, and snippets.

@etrulls
Created May 26, 2020 12:42
Show Gist options
  • Save etrulls/e37bc18411943fa54bf039013da6a88d to your computer and use it in GitHub Desktop.
Save etrulls/e37bc18411943fa54bf039013da6a88d to your computer and use it in GitHub Desktop.
root@e6c5456fa8b4:/home/trulls/src/disco-code/points# python extract_benchmark.py
[['2020-05-25/768-depth', 504, True]]
Processing /home/trulls/imw-2020/sagrada_familia and saving to out/2020-05-25/768-depth-imsize-504-nms-5/sagrada_familia
0%| | 0/100 [00:00<?, ?it/s]
Traceback (most recent call last):
File "detect.py", line 144, in <module>
described_samples = extract(extraction_method, dataset, args.h5_path)
File "detect.py", line 101, in extract
all_features = extraction_method(images.unsqueeze(0))[0]
File "detect.py", line 139, in extraction_method
return model.features(images, kind='nms')
File "/root/.local/lib/python3.8/site-packages/torch_dimcheck/dimcheck.py", line 110, in wrapped
result = func(*args, **kwargs)
File "/home/trulls/src/disco-code/points/points/model/matching_model.py", line 34, in features
features_flat = self.desc_disco.features(bitmaps_flattened, kind=kind)
File "/root/.local/lib/python3.8/site-packages/torch_dimcheck/dimcheck.py", line 110, in wrapped
result = func(*args, **kwargs)
File "/home/trulls/src/disco-code/points/points/model/desc_disco.py", line 92, in features
percepted = self.unet(images)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/root/.local/lib/python3.8/site-packages/torch_dimcheck/dimcheck.py", line 110, in wrapped
result = func(*args, **kwargs)
File "/root/.local/lib/python3.8/site-packages/unets/unet.py", line 105, in forward
features.append(layer(features[-1]))
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/container.py", line 100, in forward
input = module(input)
File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 550, in __call__
result = self.forward(*input, **kwargs)
File "/root/.local/lib/python3.8/site-packages/unets/ops.py", line 42, in forward
raise RuntimeError(msg)
RuntimeError: Trying to downsample feature map of size torch.Size([1, 64, 63, 63])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment