Welcome to fit-a-nef’s documentation!
fit-a-nef
(/fit a nɛf/) is a Python library for quick fitting of thousands of neural fields to entire datasets.
Using the ability of JAX to easily parallelize the operations on a GPU with vmap
, a sizeable set of neural
fields can be fit to distinct samples at the same time.
The fit-a-nef
library is designed to easily allow the user to add their own training task, dataset, and model.
It provides a uniform format to store and load large amounts of neural fields in a platform-agnostic way.
Whether you use PyTorch, JAX or any other framework, the neural fields can be loaded and used in your project.
Check out the Usage section for further information, including how to Install the library and the dependencies for the repository.
fit-a-nef
is developed by the Team at the University of Amsterdam.
Note
Please help us by contributing to the project! See the GitHub repository for more information.
Contents
- Usage
- Fitting Neural fields
- Hyperparameter Tuning
- API
- fit_a_nef
- fit_a_nef.initializers
- fit_a_nef.metrics
- fit_a_nef.nef
- fit_a_nef.nef.mfn
- fit_a_nef.nef.mfn.FourierNet_key
- fit_a_nef.nef.mfn.GaborNet_key
- fit_a_nef.nef.mfn.fourier_filter_linear_init
- fit_a_nef.nef.mfn.gamma_initialization
- fit_a_nef.nef.mfn.gaussian_window
- fit_a_nef.nef.mfn.simple_uniform
- fit_a_nef.nef.mfn.FourierFilter
- fit_a_nef.nef.mfn.FourierNet
- fit_a_nef.nef.mfn.GaborFilter
- fit_a_nef.nef.mfn.GaborNet
- fit_a_nef.nef.mfn.MFNBase
- fit_a_nef.nef.mlp
- fit_a_nef.nef.rffnet
- fit_a_nef.nef.siren
- fit_a_nef.nef.utils
- fit_a_nef.nef.mfn
- fit_a_nef.tasks
- fit_a_nef.tasks.image
- fit_a_nef.tasks.image.SignalImageTrainer
SignalImageTrainer
SignalImageTrainer.__init__()
SignalImageTrainer.clean_up()
SignalImageTrainer.init_model()
SignalImageTrainer.mae()
SignalImageTrainer.mse()
SignalImageTrainer.process_batch()
SignalImageTrainer.psnr()
SignalImageTrainer.simse()
SignalImageTrainer.ssim()
SignalImageTrainer.train_to_target_psnr()
SignalImageTrainer.validation_psnr()
SignalImageTrainer.verbose_train_model()
- fit_a_nef.tasks.image.SignalImageTrainer
- fit_a_nef.tasks.shape
- fit_a_nef.tasks.image
- fit_a_nef.trainer
- fit_a_nef.trainer.SignalTrainer
SignalTrainer
SignalTrainer.__init__()
SignalTrainer.apply_model()
SignalTrainer.clean_up()
SignalTrainer.compile()
SignalTrainer.create_functions()
SignalTrainer.create_train_model()
SignalTrainer.fast_train_model()
SignalTrainer.get_flat_params()
SignalTrainer.get_lr()
SignalTrainer.get_params()
SignalTrainer.init_model()
SignalTrainer.load()
SignalTrainer.process_batch()
SignalTrainer.save()
SignalTrainer.verbose_train_model()
- fit_a_nef.trainer.SignalTrainer
- fit_a_nef.utils
- fit_a_nef
- Team