Streamlit component for TensorBoard, TensorFlow's visualization toolkit

Last update: Nov 13, 2022

Overview

streamlit-tensorboard

This is a work-in-progress, providing a function to embed TensorBoard, TensorFlow's visualization toolkit, in Streamlit apps.

Installation 🎈

pip install --upgrade streamlit-tensorboard

Example Usage 💻

import streamlit as st
from streamlit_tensorboard import st_tensorboard
import tensorflow as tf

import datetime
import random

mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

def create_model():
    return tf.keras.models.Sequential(
        [
            tf.keras.layers.Flatten(input_shape=(28, 28)),
            tf.keras.layers.Dense(512, activation="relu"),
            tf.keras.layers.Dropout(0.2),
            tf.keras.layers.Dense(10, activation="softmax"),
        ]
    )

model = create_model()
model.compile(
    optimizer="adam", loss="sparse_categorical_crossentropy", metrics=["accuracy"]
)

logdir = "logs/fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=logdir, histogram_freq=1)

model.fit(
    x=x_train,
    y=y_train,
    epochs=5,
    validation_data=(x_test, y_test),
    callbacks=[tensorboard_callback],
)

# Start TensorBoard
st_tensorboard(logdir=logdir, port=6006, width=1080)

Contributing 🛠️

Please file a new GitHub issue (if one doesn't already exist) for bugs, feature requests, suggestions for improvements, etc. If you have solutions to any open issues, feel free to open a Pull Request!

Supported Platforms

Ubuntu
Debian GNU/Linux
macOS ( ⚠️ unverified)

Windows is currently not supported. PRs for added Windows support are welcome as fix to this issue.

Comments

Fixing Windows support by changing logdir to POSIX format

Using pathlib to change the logdir path to POSIX format. The change would make the shlex.split work properly, thus making it work on Windows, and it will still work on Linux.

opened by ansonnn07 2
Refuses to connect on Streamlit sharing

The issue has to do with network permissions on the remote host. Port 6006 should be opened on the remote host and incoming/outgoing connections should be allowed at remote host:6006.

opened by snehankekre 1
Works on MacOS

I read in the readme that Streamlit-tensorboard is unverified on macOS. Upon trying, I noticed a delay in the TensorBoard loading. Opening the port 6006 on another tab, helped solve this issue of the delay.

opened by 259mit 0
Support several comma-separated paths in logdir

Hi @snehankekre Many thanks for the contribution. Just wondering whether it would be possible to support passing to the logdir argument of st_tensorboard a list of comma-separated paths to render several specific experiments, e.g. in the original tensorboard call you can specify it as follows:

tensorboard --logdir=name1:/path/to/logs/1,name2:/path/to/logs/2

Regards
enhancement help wanted good first issue

opened by davidjimenezphd 1
Reuse TensorBoard on port {port} (pid {pid}) if opened previously
Each widget interaction with Streamlitt causes the script to rerun from top to bottom. This execution model leads to the creation of a new TensorBoard server for every interaction and new connection to the Streamlit app.

Desired behavior:

If a TensorBoard server is running, connect to it instead of opening a new one.

Reuse cached connection for viewers of the app. Do not open a new TensorBoard for each viewer.

bug help wanted
opened by snehankekre 3

Releases(0.0.2)

0.0.2(Jun 14, 2021)

Fixes issue #5
Source code(tar.gz)
Source code(zip)
0.0.1(Jun 13, 2021)

A lot of room for improvement! First release 0.0.1
Source code(tar.gz)
Source code(zip)

Owner

Snehan Kekre

Documentation Writer @streamlit. Formerly, @Coursera.

GitHub Repository https://pypi.org/project/streamlit-tensorboard/

Face-Recognition-based-Attendance-System - An implementation of Attendance System in python.

Face-Recognition-based-Attendance-System A real time implementation of Attendance System in python. Pre-requisites To understand the implentation of F

1 Dec 31, 2021

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

RAVE: Realtime Audio Variational autoEncoder Official implementation of RAVE: A variational autoencoder for fast and high-quality neural audio synthes

587 Jan 01, 2023

Riemannian Geometry for Molecular Surface Approximation (RGMolSA)

Riemannian Geometry for Molecular Surface Approximation (RGMolSA) Introduction Ligand-based virtual screening aims to reduce the cost and duration of

11 Nov 15, 2022

auto-tuning momentum SGD optimizer

YellowFin YellowFin is an auto-tuning optimizer based on momentum SGD which requires no manual specification of learning rate and momentum. It measure

288 Nov 19, 2022

[SIGIR22] Official PyTorch implementation for "CORE: Simple and Effective Session-based Recommendation within Consistent Representation Space".

CORE This is the official PyTorch implementation for the paper: Yupeng Hou, Binbin Hu, Zhiqiang Zhang, Wayne Xin Zhao. CORE: Simple and Effective Sess

26 Dec 19, 2022

A 10000+ hours dataset for Chinese speech recognition

WenetSpeech Official website | Paper A 10000+ Hours Multi-domain Chinese Corpus for Speech Recognition Download Please visit the official website, rea

310 Jan 03, 2023

existing and custom freqtrade strategies supporting the new hyperstrategy format.

freqtrade-strategies Description Existing and self-developed strategies, rewritten to support the new HyperStrategy format from the freqtrade-develop

39 Aug 20, 2021

Code release for Local Light Field Fusion at SIGGRAPH 2019

Local Light Field Fusion Project | Video | Paper Tensorflow implementation for novel view synthesis from sparse input images. Local Light Field Fusion

1.1k Dec 27, 2022

Code and models for "Rethinking Deep Image Prior for Denoising" (ICCV 2021)

DIP-denosing This is a code repo for Rethinking Deep Image Prior for Denoising (ICCV 2021). Addressing the relationship between Deep image prior and e

36 Dec 29, 2022

Tools for robust generative diffeomorphic slice to volume reconstruction

RGDSVR Tools for Robust Generative Diffeomorphic Slice to Volume Reconstructions (RGDSVR) This repository provides tools to implement the methods in t

0 Oct 29, 2021

Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch

PyTorch implementation of Continuous Augmented Positional Embeddings (CAPE), by Likhomanenko et al. Enhance your Transformer positional embeddings with easy-to-use augmentations!

26 Dec 13, 2022

ThunderSVM: A Fast SVM Library on GPUs and CPUs

What's new We have recently released ThunderGBM, a fast GBDT and Random Forest library on GPUs. add scikit-learn interface, see here Overview The miss

1.4k Dec 22, 2022

The GitHub repository for the paper: “Time Series is a Special Sequence: Forecasting with Sample Convolution and Interaction“.

SCINet This is the original PyTorch implementation of the following work: Time Series is a Special Sequence: Forecasting with Sample Convolution and I

386 Jan 01, 2023

Streamlit component for TensorBoard, TensorFlow's visualization toolkit

Related tags

Overview

streamlit-tensorboard

Installation 🎈

Example Usage 💻

Contributing 🛠️

Supported Platforms

Comments

Fixing Windows support by changing logdir to POSIX format

Refuses to connect on Streamlit sharing

Works on MacOS

Support several comma-separated paths in logdir

Reuse TensorBoard on port {port} (pid {pid}) if opened previously

Releases(0.0.2)

0.0.2(Jun 14, 2021)

0.0.1(Jun 13, 2021)

Owner

Snehan Kekre

Face-Recognition-based-Attendance-System - An implementation of Attendance System in python.

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

Riemannian Geometry for Molecular Surface Approximation (RGMolSA)

auto-tuning momentum SGD optimizer

[SIGIR22] Official PyTorch implementation for "CORE: Simple and Effective Session-based Recommendation within Consistent Representation Space".

A 10000+ hours dataset for Chinese speech recognition

existing and custom freqtrade strategies supporting the new hyperstrategy format.

Code release for Local Light Field Fusion at SIGGRAPH 2019

Code and models for "Rethinking Deep Image Prior for Denoising" (ICCV 2021)

Tools for robust generative diffeomorphic slice to volume reconstruction

Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch

ThunderSVM: A Fast SVM Library on GPUs and CPUs

The GitHub repository for the paper: “Time Series is a Special Sequence: Forecasting with Sample Convolution and Interaction“.

Python tools for 3D face: 3DMM, Mesh processing(transform, camera, light, render), 3D face representations.

AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation

A DCGAN to generate anime faces using custom mined dataset

Pre-trained models for a Cascaded-FCN in caffe and tensorflow that segments

Alpha-Zero - Telegram Group Manager Bot Written In Python Using Pyrogram

Code for ICDM2020 full paper: "Sub-graph Contrast for Scalable Self-Supervised Graph Representation Learning"

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners