search

LIAGM / LFattNet

Attention-based View Selection Networks for Light-field Disparity Estimation
MIT License
54 stars 15 forks source link

How to imlement operation? #4

Closed dreamhua82 closed 4 years ago

dreamhua82 commented 4 years ago

Dear LIAGM, Can the code below implement the operation? is it right ? I use this in the program, but find the Cost volume became Nan !

Thanks

LIAGM commented 4 years ago

Hi

I just wonder why you want to use correlation in the costvolume? And as I remembered, this part of the code is to construct the costvolume. The tmp_list contains N*N shifted feature map, and I am not sure what you want to correlate with? Can you describe more clearly for the output of the function?

Regards,

LIAGM commented 4 years ago

Hi

I am not sure that the function of multiply can do the operation you want. Maybe you can simply do some correlation operation in the channel dimension separately to make it become a scalar. These are the suggestion I can give to you

Regards,