This packages contains frequently used methods or classes in use of Tensorflow 2.x
Project description
tfhelper
Your friendly Tensorflow 2.x neighbor.
Documentation: https://jeikeilim.github.io/tfhelper/
Environments
- Python 3.7
Getting started
pip install
pip install tfhelper
Structure
tfhelper.dataset
- HDF5Generator
from tfhelper.dataset import HDF5Generator
train_generator = HDF5Generator("/dataset/path.hdf5", "training_data", "test_data")
train_dataset = train_generator.get_dataset(input_shape=(300, 300), batch_size=16, shuffle=True, n_shuffle=1000)
...
model = Define some model
model.fit(train_dataset)
tfhelper.gpu
- allow_gpu_memory_growth
tfhelper.tensorboard
- ConfuseCallback
- ModelSaverCallback
- SparsityCallback
- run_tensorboard
- wait_ctrl_c
- get_tf_callbacks
tfhelper.tflite
- keras_model_to_tflite
- parse_config
- predict_tflite_interpreter
- evaluate_tflite_interpreter
- load_pruned_model
tfhelper.transfler_learning
- get_transfer_learning_model
tfhelper.visualization
- get_cam_image
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
tfhelper-0.0.33.tar.gz
(13.7 kB
view hashes)
Built Distribution
tfhelper-0.0.33-py3-none-any.whl
(17.7 kB
view hashes)
Close
Hashes for tfhelper-0.0.33-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8ffff52f7417666416763ccacdbe92bac4b7c160b256366ee6ef00f1fbc5d4b9 |
|
MD5 | bb4d846143fa2e3502c5e744d7ec20e4 |
|
BLAKE2b-256 | 1ae5b511e38c1e56b3c6c8cb7d17b474a043006d7fa55a0aee4db3ced7dc949e |