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
- Clone this repo
git clone https://github.com/taipingeric/fusionlab.git
cd fusionlab
- (remove anaconda first)
- Install Miniconda
- Miniconda3 macOS Apple M1 64-bit pkg
- Miniconda3 macOS Apple M1 64-bit bash
- Install the xcode-select command-line
xcode-select --install
- Deactivate the base environment
conda deactivate
- Create conda environment using config
conda env create -f ./configs/tf-apple-m1-conda.yaml -n fusionlab
- Replace requirements.txt with requirements-m1.txt
- Install by pip
pip install -r requirements-m1.txt
How to use
import fusionlab as fl
# PyTorch
encoder = fl.encoders.VGG16()
# Tensorflow
encoder = fl.encoders.TFVGG16()
Encoders
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)
- UNet, TFUNet
- ResUNet, TFResUNet
- UNet2plus, TFUNet2plus
Project details
Release history Release notifications | RSS feed
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.34.tar.gz
(16.5 kB
view hashes)
Built Distribution
fusionlab-0.0.34-py3-none-any.whl
(27.0 kB
view hashes)
Close
Hashes for fusionlab-0.0.34-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 225012c367504bef26c7b92d86759a39405574d2b01d9309dd5c8f2d5c6d3d1a |
|
MD5 | 03e97f1484ea8ca8a3ebdec5df0625b1 |
|
BLAKE2b-256 | ea51aaea5925a1fe4dc0d5ea7094b2cfa7f1db9c8a86428a4f7291bec07d8b79 |