Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MAnet #310

Merged
merged 4 commits into from
Dec 15, 2020
Merged

Add MAnet #310

merged 4 commits into from
Dec 15, 2020

Conversation

azkalot1
Copy link
Contributor

Paper: https://ieeexplore.ieee.org/abstract/document/9201310
Seems to work better than Unet, similar to Unet++ but faster

An implementation is similar to the original in the paper, except for encoder part.

@qubvel
Copy link
Collaborator

qubvel commented Dec 15, 2020

A few more things:

  • Update README.md
  • Update docs/models.rst
  • Check new model docstring to be constistent with existing model docstrings

May be write a little bit more in description of the model in docstring

README.md Outdated Show resolved Hide resolved
segmentation_models_pytorch/__init__.py Outdated Show resolved Hide resolved
segmentation_models_pytorch/manet/model.py Show resolved Hide resolved
segmentation_models_pytorch/manet/model.py Outdated Show resolved Hide resolved
@qubvel
Copy link
Collaborator

qubvel commented Dec 15, 2020

👍

@qubvel qubvel merged commit a4220a0 into qubvel-org:master Dec 15, 2020
yuki-inaho pushed a commit to yuki-inaho/segmentation_models.pytorch that referenced this pull request Dec 16, 2020
* add MAnet arch

* add MAnet arch

* update docs, readme, docstring

* fix docstring, rename decoder parameters
@ynhuhu
Copy link

ynhuhu commented Feb 27, 2021

Thanks sharing!
I installed segmentation_models_pytorch
model = smp.MAnet(encoder_name=ENCODER, encoder_weights=ENCODER_WEIGHTS,
classes=len(CLASSES),
activation=ACTIVATION,
)

module 'segmentation_models_pytorch' has no attribute 'MAnet'
Why?

@qubvel
Copy link
Collaborator

qubvel commented Feb 27, 2021

Install the latest version, look at the instructions in README.md

@pritesh-aidash
Copy link

I have the latest version. MAnet still does not show up.

@felipemello1
Copy link

@pritesh-aidash @qubvel same thing for me.

So I did

import segmentation_models_pytorch as smp
smp??

And it showed

from .unet import Unet
from .unetplusplus import UnetPlusPlus
from .manet import MAnet
from .linknet import Linknet
from .fpn import FPN
[...]

trying to do smp.MAnet doesnt work, but smp.manet.MAnet does

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants