YimianDai / open-aff

code and trained models for "Attentional Feature Fusion"
729 stars 95 forks source link

Do you have the code of Pytorch version? #2

Closed ma3252788 closed 3 years ago

ma3252788 commented 4 years ago

Or will you consider publishing the pytorch version of the code in the future?

YimianDai commented 4 years ago

I am going to convert to PyTorch since it is the most dominant framework. It is a bit meaningless to release the code in a framework that few people are using, although I like MXNet very much.

But not in the near future. I am in quite a chaos and uncertain status recently.

If everything is set, I am planning to re-implement it in PyTorch in the coming spring festival.

ma3252788 commented 4 years ago

I am going to convert to PyTorch since it is the most dominant framework. It is a bit meaningless to release the code in a framework that few people are using, although I like MXNet very much.

But not in the near future. I am in quite a chaos and uncertain status recently.

If everything is set, I am planning to re-implement it in PyTorch in the coming spring festival.

Okay, thank you very much, I look forward to seeing the pytorch version of aff in the near future~

majunfu commented 4 years ago

Waiting for your pytorch version

Senwang98 commented 3 years ago

@YimianDai Waiting for your pytorch version

YimianDai commented 3 years ago

@1187697147 Hi, @bobo0810 has contributed a Pytorch version https://github.com/YimianDai/open-aff/tree/master/aff_pytorch. Although it is not written by me, I hope it will help you.

As to me, currently, I am busy with finalizing my PhD Oral defense. I think I will have a lot of free time next week (I hope so) and will start to do all the things that I have ever promised in my replied issues of this repo.

Sorry for the delay.

Senwang98 commented 3 years ago

@YimianDai Nice, good luck to you and thanks for your work!

bobo0810 commented 3 years ago

Description:

  1. MS_CAM, AFF, iAFF fusion operation are completely consistent with the MXNet version.
  2. AFF-ResNet is Pytorch official ResNet + AFF/iAFF.

Good luck!:smiley:

hiyyg commented 3 years ago

@bobo0810 @YimianDai Thanks for your great work, do you have pretrained models for pytorch version?

bobo0810 commented 3 years ago

@bobo0810 @YimianDai Thanks for your great work, do you have pretrained models for pytorch version?

Unfortunately:pensive:, I may not be able to provide it. Because I only apply it to the company's private data set.

YimianDai commented 3 years ago

@hiyyg Hi, many thanks for being interested in my work. I am considering to fully convert to PyTorch and extend the conference paper to a journal article with more experiments.

If the accuracy of my new PyTorch implementation (not done yet) does not drop, I will release the pretrained models for PyTorch version. Please stay patient. I will update the process in this issue in the following days.

ma3252788 commented 3 years ago

@hiyyg Hi, many thanks for being interested in my work. I am considering to fully convert to PyTorch and extend the conference paper to a journal article with more experiments.

If the accuracy of my new PyTorch implementation (not done yet) does not drop, I will release the pretrained models for PyTorch version. Please stay patient. I will update the process in this issue in the following days.

What's the result, please?

YimianDai commented 3 years ago

I am working on this. The next version will be PyTorch and Open-MMLab based.

ma3252788 commented 3 years ago

I am working on this. The next version will be PyTorch and Open-MMLab based.

Got it. Thank you for your reply. Looking forward to your work

fhaji75 commented 9 months ago

@hiyyg Hi, many thanks for being interested in my work. I am considering to fully convert to PyTorch and extend the conference paper to a journal article with more experiments.

If the accuracy of my new PyTorch implementation (not done yet) does not drop, I will release the pretrained models for PyTorch version. Please stay patient. I will update the process in this issue in the following days.

Hello, thanks for your good repository, Can you release pre-trained models for Pytorch?

CaoZhonglei commented 9 months ago

请问有tensorflow版本的代码吗?