Hi,
Thank you for sharing your code.
I think there is a bug in
|
self.feature_extractor = ExtractFeatureMap(im_fe_ratio=self.im_fe_ratio, device=postprocess_device) |
|
self.postprocess_device = postprocess_device |
|
self.im_fe_ratio = im_fe_ratio |
L144 should be move to after L146.
Thank you.