Skip to content

brookisme/tfbox

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

264 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

tfbox

a collection of models and tools for tensorflow


TFBox's main utility lies within:

Additionally, TFBox contains a number of useful tools for TensorFlow, including:


INSTALL

pip install tfbox

ADDITIONAL REQUIREMENTS


MODELS

tfbox.nn.encoder/decoder/encoder-decoder use yaml files to combine keras-model-blocks in tfbox.nn.blocks to build neural-networks. The result is an flexible system from which you can build a large variety of models. Lets start with some examples.

Here is the config for the Xception Network:

xception:

    blocks_config:
        - conv:
            filters: 32
            strides: 2
        - 64
        - stack:
            name: entry_flow_blocks
            seperable: true
            depth: 3
            output_stride: 2
            layers: [128,256,728]
        - stack:
            name: middle_flow
            nb_repeats: 16
            depth: 3
            filters: 728
        - stack:
            name: exit_flow_block
            output_stride: 2
            filters_list: [728,1024,1024]
        - stack:
            name: exit_flow_convs
            seperable: true
            residual: false
            layers: [1536,1536,2048]
        - aspp


DFSequence

tfbox.loaders.DFSequence builds instances of tf.keras.utils.Sequence for image segmentation models using pandas dataframes. In particular it does almost anything you can imagine - but also can be bit overwhelming.


SCORING


METRICS


LOSS FUNCTIONS


TENSORBOARD CALLBACKS

About

Tensorflow Models and Tools

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages