search

Megvii-BaseDetection / DisAlign

Implementation of "Distribution Alignment: A Unified Framework for Long-tail Visual Recognition"(CVPR 2021)
Apache License 2.0
117 stars 10 forks source link

Where is the DisAlignLinear module? #21

Closed jeongHwarr closed 2 years ago

jeongHwarr commented 2 years ago

Hello. Thank you for your impressive project!

I want to apply DisAlign to classification. However, an error occurs in the import part. https://github.com/Megvii-BaseDetection/DisAlign/blob/a2fc3500a108cb83e3942293a5675c97ab3a2c6e/classification/imagenetlt/resnext50/resx50.scratch.imagenet_lt.224size.90e.disalign.10e/net.py#L7 I coudn't find the DisAlignLinear in cvpods.layers. and there also isn't exist at https://github.com/Megvii-BaseDetection/cvpods/tree/master/cvpods/layers How can I solve this problem?

Thank you!

FateScript commented 2 years ago

It's here Sorry, it's a bug in cvpods, I will fix this later. You could use from cvpods.layers.wrappers import DisAlignLinear to work around this.

jeongHwarr commented 2 years ago

@FateScript Thank you so much!

FateScript commented 2 years ago

fixed now @jeongHwarr , please pull our latest branch.

jeongHwarr commented 2 years ago

@FateScript Thank you!