tensorboard_image_extractor/README.md

79 lines
2.2 KiB
Markdown
Raw Normal View History

2021-06-09 16:55:29 +02:00
# Tensorboard Image Extractor
## What is this program and why?
2021-06-09 16:34:44 +02:00
2021-06-09 16:55:29 +02:00
This is short script to extract images and create animated gif from there
using a tensorboard event file. Unfortunatly this feature is not native inside
tensorboard, inside which only the graphing data can downloaded (in `csv` or
`json` format).
The only other program which I found that did a similar thing is
https://github.com/lanpa/tensorboard-dumper/ which I took inspiration from.
## How to use it
The repository can be clone with git and the you will maybe need to install
some dependencies (like tensorboard):
```
pip3 install -r requirements.txt
```
You can then run it:
```
python3 tensorboard_image_extractor.py -i event.db
```
You can get some help by running:
```
python3 tensorboard_image_extractor.py --help
```
## Tensorboard datastructure
The following diagram describes a tree of the log directory found in all
machine learning experiment with a tensorboard writer.
```
logs/
├── lupo
│   ├── args.txt
│   ├── config.txt
│   ├── model_005000.pth
│   ├── model_010000.pth
│   ├── model_015000.pth
│   ├── model_020000.pth
│   └── model_025000.pth
└── summaries
└── lupo
└── events.out.tfevents.1623155921.pop-os
```
The file which contains all data and images is the `event` file in
`logs/summaries/{run name}/events`. It can be fairly large because every image
is stored inside in binary format.
## Example
You can create an animated gif, only keeping images with a certain `tag`:
```
python3 tensorboard_image_extractor.py -i lupo.events -t "train/level_1/rgb" -o train_level_1_rgb_24h.gif --gif
```
## Performance
In order to create a gif from a 900 MB event file, it took me just over an
hour. This is due to the fact that Python has to do the I/O reading from binary
data and converting the whole file, which is remarkably slow.
It can create large gif files. In the experiment described above the images of
a single tag was kept and it created a 52 MB gif file.
## Notes
This program is distributed under GNU GNL v3 or later License, which you can
find a copy of in the repository.
This program comes with ABSOLUTELY NO WARRANTY
Tensorboard Image Extractor - Copyright (C) 2021 - Otthorn