tf-slim

What it is and what it does

TensorFlow-Slim is a lightweight library that sits on top of TensorFlow to reduce boilerplate when building neural networks. It provides argument scoping to set default parameters across operations, high-level layer definitions, variable management utilities that distinguish between model and non-model variables, and pre-built implementations of popular computer vision architectures like VGG and AlexNet. It also includes training and evaluation routines, loss functions, metrics, regularizers, and data-loading utilities.

You use it by importing as `tf_slim` and calling its layers and utilities instead of writing raw TensorFlow operations. It is designed to mix freely with native TensorFlow code, so you can adopt it incrementally. The library targets developers building convolutional neural networks and other deep learning models who want to reduce repetitive code and improve readability.

Use it for

• Define convolutional neural networks compactly using slim layers instead of native operations
• Reuse pre-trained VGG or AlexNet models as starting points or feature extractors for transfer learning
• Manage model variables separately from training variables for easier checkpoint saving and loading
• Apply regularization and common loss functions without writing custom code
• Train models with built-in learning routines and evaluate using standard metrics

Worth the install?

AI-flagged interpretation of the facts on this page. Verify before relying on it.

Install

tf-slim on PyPI

Before you install

Low friction to install; depends only on absl-py. Maintenance is active with recent commits, though the latest release was 2020-05-07 and the package has not received updates for several years despite the repository remaining active.

Requires TensorFlow 1.15.2 or TensorFlow 2.0–2.2; compatibility with TensorFlow versions beyond 2.2 is unverified.

License in practice

Licensed under Apache 2.0 (permissive), allowing free use, modification, and distribution with minimal restrictions—suitable for both open-source and commercial projects.

Quickstart

pip install tf-slim

import tf_slim as slim

# Define a model using slim layers and variables
weights = slim.model_variable('weights', shape=[10, 10, 3, 3], initializer=tf.truncated_normal_initializer(stddev=0.1), regularizer=slim.l2_regularizer(0.05))

Package facts

Evidence: tf_slim-1.1.0-py2.py3-none-any.whl

Tags

More Software Development packages

See also tf-keras-nightly · thinc · tensorflow · tensorflow-addons · tensorflow-graphics · tf-keras · tensorflow-recommenders · tf2crf · docker-squash · NeuralFoil