Skip to content

Base code for running train and evaluation models, both on local and distributed config

Notifications You must be signed in to change notification settings

medtune/Models-implementation

Repository files navigation

Models-implementation

This is an implementation of CNN's architectures using TensorFlow. The main goal of this project is to support the creation of machine learning & deep learning Pipelines .

lien

Description

The repository is composed of:

  • Scripts for launching training and evaluation: It includes the basic tf.Session() and tf.train.MonitoredSession with different hooks and scaffold for distributed training.
  • Folder Densenet: Regroup image preprocessing functions, script for a slim-like implementation of Densenet
  • Folder utils: Convert data to Tfrecord format, analyse your data depending on its nature

##Installation

First, install Tensorflow (Or/And tensorflow-gpu) in order to perform computation in cpu-only (In GPU) Then, in order to use "slim" package, which is developped under : lien, git clone the above repository. Copy the "research" folder and perform both setup.py :

  • under research folder
  • under slim folder (make sure you delete BUILD file before running setup.py)

As a last step, git clone Models-implementation repo : https://www.github.com/medtune/Models-implementation

Usage example

Release History

About

Base code for running train and evaluation models, both on local and distributed config

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages