Skip to main content

Useful packages for DL

Project description

FusionLab



FusionLab is an open-source frameworks built for Deep Learning research written in PyTorch and Tensorflow. The code is easy to read and modify especially for newbie. Feel free to send pull requests :D

Installation

With pip

pip install fusionlab

For Mac M1 chip users

Requirements:

  • Apple Mac with M1 chips
  • MacOS > 12.6 (Monterey)

Following steps

  1. Clone this repo
git clone https://github.com/taipingeric/fusionlab.git
cd fusionlab
  1. (remove anaconda first)
  2. Install Miniconda
    1. Miniconda3 macOS Apple M1 64-bit pkg
    2. Miniconda3 macOS Apple M1 64-bit bash
  3. Install the xcode-select command-line
xcode-select --install
  1. Deactivate the base environment
conda deactivate 
  1. Create conda environment using config
conda env create -f ./configs/tf-apple-m1-conda.yaml -n fusionlab
  1. Replace requirements.txt with requirements-m1.txt
  2. Install by pip
pip install -r requirements-m1.txt

ref: https://github.com/jeffheaton/t81_558_deep_learning/install/tensorflow-install-conda-mac-metal-jan-2023.ipynb

video

How to use

import fusionlab as fl

# PyTorch
encoder = fl.encoders.VGG16()
# Tensorflow
encoder = fl.encoders.TFVGG16()

Encoders

encoder list

Segmentation

import fusionlab as fl

# Segmentation Model

# PyTorch UNet
unet = fl.segmentation.UNet(cin=3, num_cls=10, base_dim=64)
# Tensorflow UNet
unet = fl.segmentation.TFUNet(num_cls=10, base_dim=64)

Segmentation model list

  • UNet, TFUNet
  • ResUNet, TFResUNet
  • UNet2plus, TFUNet2plus

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

fusionlab-0.0.35.tar.gz (15.1 kB view hashes)

Uploaded Source

Built Distribution

fusionlab-0.0.35-py3-none-any.whl (29.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page