Skip to content
This repository has been archived by the owner on Oct 25, 2023. It is now read-only.

sunziping2016/transfer-tensorflow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

59 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Table of contents

Xlearn Transfer Learning Library

The Xlearn transfer learning library implements some tranfer learning or domain adaption algorithms using TensorFlow. The script main.py is for both training and evaluating the models. And it also contains some scripts for downloading and preprocessing the data and the pretrained models.

This library is still being heavily developed. Vast changes are expected these days.

Setup

This library is Python 2&3 compatible. See requirements.txt for the detailed dependencies. You may modify it if you want to use the gpu version of TensorFlow.

Usage

Quick Start

Excute the following script can run Alexnet-based DAN models on the Office dataset.

# Prepare the data
sh data/office/download.sh
# Download and process the pretrained mean
python download_mean_from_caffe.py
# Download and process the pretrained model
python download_model_from_caffe.py
# Train
python main.py

Methods

Currently it only supports DAN with Caffe alexnet base-model.

About

A TensorFlow Library for Transfer Learning

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published