Skip to main content

a simple package for handling tensorflow tensor

Project description

README

NEWS

Date News Version
August 2019 deeplab semantic segmentation (preview version)was release >= v0.6.10
Mei 2019 yolov3 (stable version) was relesed > v0.5.1
Mei 2019 U-net semantic segmentation (stable version) was released > v0.5.1
April 2019 yolov3 (preview version) v0.4.18
April 2019 Unet-segmentation (preview version) v0.4.18

Tensorflow Compatibility

Tensorflow version Simple-Tensor Version
1.4.1 - 1.12 >=v0.4.0
1.13.1 >=v0.4.3

ABOUT PROJECT

This project is a simplification of tensorflow operations and related projects

DEPENDENCIES

  1. Tensorflow (1.4.1 - 1.13)

For installing tensorflow, with GPU:

# python3 
pip3 install tensorflow-gpu
# python2
pip2 install tensorflow-gpu

Without GPU:

# python3 
pip3 install tensorflow
# python2
pip2 install tensorflow

HOW TO USE

:shipit: Installing The Package

python setup.py install

or

pip3 install simple-tensor

:shipit: Import The Package

Tensor Operations

import tensorflow as tf
# tensor operations
from simple_tensor.tensor_operations import *
# tensor losses
from simple_tensor.tensor_losses import *
# tensor metrics
from simple_tensor.tensor_metrics import *

This packages contains tensor operation (conv2d, conv1d, depthwise conv2d, fully connected, conv2d transpose), tensor losses (softmax & sigmoid cross entropy, MSE), and tensor metrics (accuracy). For more detail documentations about tensor operations, visit this page

Convert Keras Model to Tensorflow Serving

import tensorflow as tf
from simple_tensor.convert import *

Transfer Learning Package

import tensorflow as tf
from simple_tensor.transfer_learning.inception_utils import *
from simple_tensor.transfer_learning.inception_v4 import *

This package contains a library of tensorflow implementation of Inception-v4 for image classification. Densenet, Resnet, and VGG will be added in the future version. For more detail documentations about transfer learning package, visit this page

alt text

(img source: link)

Object Detector Package

import tensorflow as tf
from simple_tensor.object_detector.detector_utils import *
from simple_tensor.object_detector.yolo_v4 import *

This package contains a library of tensorflow implementation of Yolov3 (training and inferencing). You can customize your yolo detector with four types of network ("big", 'medium", "small", "very_small"). For more detail documentations about object detector package (yolov3), visit this page.

alt text

(img source: pjreddie)

Unet Segmentation Package

import tensorflow as tf
from simple_tensor.segmentation.unet import UNet

This package contains the tensorflow implementation of U-net for semantic segmentation. For more detail, visit this page

alt text

(img source: internal)

LSTM Package

still on progress ....

DOCKER

We already prepared the all in one docker for computer vision and deep learning libraries, including tensorflow 1.12, Opencv3.4.2 and contrib, CUDA 9, CUDNN 7, Keras, jupyter, numpy, sklearn, scipy, statsmodel, pandas, matplotlib, seaborn, flask, gunicorn etc. See the list of dockerfile below:

Docker: Ubuntu 16.04 with GPU (Cuda 9, cudnn 7.2) [TESTED]
Docker: Ubuntu 18.04 with GPU (Cuda 9, cudnn 7.2)
Docker: Ubuntu 16.04 without GPU (Cuda 9, cudnn 7.2) [TESTED]
Docker: Ubuntu 18.04 without GPU (Cuda 9, cudnn 7.2) [TESTED]

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 Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

simple_tensor-0.6.13-py3.5.egg (94.0 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page