Created
May 26, 2020 12:42
-
-
Save etrulls/e37bc18411943fa54bf039013da6a88d to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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