search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.07k stars 181 forks source link

Dimension mismatch #9

Closed ajundo closed 3 years ago

ajundo commented 3 years ago

Nice job!

I'm using r2cnn detector. When I adjust the BASE_ANCHOR_SIZE_LIST and ANCHOR_STRIDE for small object detection, the follow error happens:

when BASE_ANCHOR_SIZE_LIST = [32,64,128] and ANCHOR_STRIDE = [4, 8, 16]: (0) Invalid argument: Incompatible shapes: [367500] vs. [372883] [[node tower_0/postprocess_FPN/mul_3 (defined at ../../libs/utils/bbox_transform.py:29) ]]

when BASE_ANCHOR_SIZE_LIST = [32,64,128, 256] and ANCHOR_STRIDE = [4, 8, 16, 32]: (0) Invalid argument: Incompatible shapes: [371875] vs. [372883] [[node tower_0/postprocess_FPN/mul_3 (defined at ../../libs/utils/bbox_transform.py:29) ]]

Do I miss some other necessary modification?