-
Notifications
You must be signed in to change notification settings - Fork 7
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
NF: Autoencoder FINTA ConvNext #246
base: master
Are you sure you want to change the base?
NF: Autoencoder FINTA ConvNext #246
Conversation
… into add_autoencoder_streamlines
Hello @AntoineTheb, Thank you for updating ! There are currently no PEP 8 issues detected in this Pull Request. Cheers! 🍻 Comment last updated at 2024-10-03 21:03:57 UTC |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm !
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM - Either we wait for latent space viz before merging or Jeremi or I could add it after merging.
@@ -0,0 +1,136 @@ | |||
#!/usr/bin/env python | |||
# -*- coding: utf-8 -*- |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Rename to ae_autoencode_tractogram.py
"as a data per point (color)") | ||
|
||
# Additional arg for projects | ||
p.add_argument('--model', type=str, choices=['finta', 'convnext'], |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Put it as a mandatory argument.
@arnaudbore don't wait for me I'll be busy with MICCAI then vacations so I'll rebase and adjust once @levje's pr is merged |
Description
Add a new autoencoder model inspired by ConvNext: https://arxiv.org/abs/2201.03545
Testing data and script
Ask Arnaud for data
Have you
People this PR concerns