Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/dalmia/siren
PyTorch implementation of Sinusodial Representation networks (SIREN)
https://github.com/dalmia/siren
activation-functions deep-learning machine-learning pytorch
Last synced: about 21 hours ago
JSON representation
PyTorch implementation of Sinusodial Representation networks (SIREN)
- Host: GitHub
- URL: https://github.com/dalmia/siren
- Owner: dalmia
- License: mit
- Created: 2020-06-19T21:13:24.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2022-12-08T06:21:22.000Z (almost 2 years ago)
- Last Synced: 2024-10-30T21:02:46.642Z (15 days ago)
- Topics: activation-functions, deep-learning, machine-learning, pytorch
- Language: Python
- Homepage:
- Size: 6.97 MB
- Stars: 263
- Watchers: 8
- Forks: 11
- Open Issues: 18
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Sinusoidal Representation Networks (SIREN)
Unofficial PyTorch implementation of Sinusodial Representation networks (SIREN) from the paper [Implicit Neural Representations with Periodic Activation Functions](https://arxiv.org/abs/2006.09661). This repository is a PyTorch port of [this](https://github.com/titu1994/tf_SIREN) excellent TF 2.0 implementation of the same.
If you are using this codebase in your research, please use the following citation:
```
@software{aman_dalmia_2020_3902941,
author = {Aman Dalmia},
title = {dalmia/siren},
month = jun,
year = 2020,
publisher = {Zenodo},
version = {v1.1},
doi = {10.5281/zenodo.3902941},
url = {https://doi.org/10.5281/zenodo.3902941}
}
```
# Setup
- Install using pip
```
$ pip install siren-torch
```
# Usage
## Sine activation
You can use the `Sine` activation as any other activation
```python
from siren import Sine
x = torch.rand(10)
y = Sine(w0=1)(x)
```
## Initialization
The authors in the paper propose a principled way of intializing the layers for the SIREN model. The initialization function
can be used as any other initialization present in `torch.nn.init`.
```python
from siren.init import siren_uniform_
w = torch.empty(3, 5)
siren_uniform_(w, mode='fan_in', c=6)
```
## SIREN model
The SIREN model used in the paper, with sine activation and custom initialization, can directly be created as follows.
```python
from siren import SIREN
# defining the model
layers = [256, 256, 256, 256, 256]
in_features = 2
out_features = 3
initializer = 'siren'
w0 = 1.0
w0_initial = 30.0
c = 6
model = SIREN(
layers, in_features, out_features, w0, w0_initial,
initializer=initializer, c=c)
# defining the input
x = torch.rand(10, 2)
# forward pass
y = model(x)
```
# Results on Image Inpainting task
A partial implementation of the image inpainting task is available as the `train_inpainting_siren.py` and `eval_inpainting_siren.py` scripts.
To run training:
```bash
$ python scripts/train_inpainting_siren.py
```
To run evaluation:
```bash
$ python scripts/eval_inpainting_siren.py
```
Weight files are made available in the repository under the `checkpoints` directory. It generates the following output after 5000 epochs of training with batch size 8192 while using only 10% of the available pixels in the image during training phase.
# Tests
Tests are written using `unittest`. You can run any script under the `tests` folder.
# Contributing
As mentioned at the beginning, this codebase is a PyTorch port of [this](https://github.com/titu1994/tf_SIREN). So, I might have missed a few details mentioned in the original paper. Assuming that the implemention in the linked repo is correct, one can safely trust this implementation as well. The only major difference from the reference repo is that it has `w0` as part of the initialization as well. I did not see that in the paper and hence, didn't include it here. I have not deeply read the paper and this is simply to serve as a starting point for anyone looking for the implementation. Please feel free to make a PR or create an issue if you find a bug or you want to contribute to improve any other aspect of the codebase.