mihaidusmanu / d2-net

D2-Net: A Trainable CNN for Joint Description and Detection of Local Features
Other
782 stars 164 forks source link

An error occurred when extracting multiscale features #29

Closed stx024 closed 4 years ago

stx024 commented 4 years ago

When I try to extract multiscale features, an error occurred like this:

Traceback (most recent call last): File "extract_features.py", line 120, in model File "/data1/stx/D2-net/d2-net-master/lib/pyramid.py", line 46, in process_multiscale detections = torch.min(detections, (1 - banned)) RuntimeError: Expected object of scalar type Bool but got scalar type Byte for argument #2 'other' in call to _th_min

My python version is 3.6.6, pytorch version is 1.3.0

mihaidusmanu commented 4 years ago

You are not using the latest version of the D2-Net code. This was fixed in the latest commit. Make sure to pull before running the feature extraction again. Leaving this issue open in the meanwhile.

stx024 commented 4 years ago

You are not using the latest version of the D2-Net code. This was fixed in the latest commit. Make sure to pull before running the feature extraction again. Leaving this issue open in the meanwhile.

Thank you very much!