├── .github └── workflows │ └── ci.yml ├── .gitignore ├── .style.yapf ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── examples └── transfer_learning.ipynb ├── flaxvision ├── __init__.py ├── models │ ├── __init__.py │ ├── densenet.py │ ├── inception.py │ ├── resnet.py │ ├── segmentation │ │ ├── __init__.py │ │ ├── deeplabv3.py │ │ ├── fcn.py │ │ └── segmentation.py │ └── vgg.py └── utils.py ├── setup.py └── tests ├── __init__.py ├── run_tests.sh ├── test_models.py ├── test_pretrained.py └── test_training.py /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/.gitignore -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/.style.yapf -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/README.md -------------------------------------------------------------------------------- /examples/transfer_learning.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/examples/transfer_learning.ipynb -------------------------------------------------------------------------------- /flaxvision/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/__init__.py -------------------------------------------------------------------------------- /flaxvision/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/__init__.py -------------------------------------------------------------------------------- /flaxvision/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/densenet.py -------------------------------------------------------------------------------- /flaxvision/models/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/inception.py -------------------------------------------------------------------------------- /flaxvision/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/resnet.py -------------------------------------------------------------------------------- /flaxvision/models/segmentation/__init__.py: -------------------------------------------------------------------------------- 1 | from .segmentation import * 2 | -------------------------------------------------------------------------------- /flaxvision/models/segmentation/deeplabv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/segmentation/deeplabv3.py -------------------------------------------------------------------------------- /flaxvision/models/segmentation/fcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/segmentation/fcn.py -------------------------------------------------------------------------------- /flaxvision/models/segmentation/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/segmentation/segmentation.py -------------------------------------------------------------------------------- /flaxvision/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/models/vgg.py -------------------------------------------------------------------------------- /flaxvision/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/flaxvision/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/run_tests.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/tests/run_tests.sh -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_pretrained.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/tests/test_pretrained.py -------------------------------------------------------------------------------- /tests/test_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rolandgvc/flaxvision/HEAD/tests/test_training.py --------------------------------------------------------------------------------