Official Pytorch implementation of Meta Internal Learning

Last update: Aug 24, 2022

Related tags

Overview

Meta Internal Learning

This repository is the official implementation of Meta Internal Learning by Raphael Bensadoun, Shir Gur, Tomer Galanti, Lior Wolf.

Project | arXiv | Code

Requirements

To install requirements:

pip install -r requirements.txt

Training on a dataset

Create a folder X containing the images. (see structure in data folder)
Determine how many iterations Y to train by scale (depends mostly on the size of the dataset, you may refer to the appendix for reference).
Run

python train.py --image-path X --batch-size 16 --visualize --niter Y --min-size 25 --checkname X_result --SAVE-MODEL --SAVE-IMGS

Generated images, tensorboard logs and trained models are stored in MetaInternalLearning/run/X_result.

The default input format is jpg. Use '--file-suffix png' for .png files.

Examples from paper -

Places-50 -

python train.py --image-path data/places_50 --batch-size 16 --visualize --niter 4000  --min-size 28 --checkname places_50_result --SAVE-MODEL --SAVE-IMGS

LSUN-50

python train.py --image-path data/lsun_50 --batch-size 16 --visualize --niter 5000  --checkname lsun_50_result --SAVE-MODEL --SAVE-IMGS

Valley dataset can be downloaded here - http://places2.csail.mit.edu/download.html (256x256 small images) and can be divided into subsets as mentioned in the paper.

V500 -

python train.py --image-path data/V500 --batch-size 16 --visualize --niter 25000 --min-size 25 --checkname v500_result --ar 1 --SAVE-MODEL

V2500 -

python train_dataset_parallel.py --image-path data/V2500 --batch-size 16 --niter 100000 --rec-weight 50 --min-size 25 --checkname v2500_result --ar 1 --SAVE-MODEL

V5000 -

python train_dataset_parallel.py --image-path data/V5000 --batch-size 16 --niter 150000 --rec-weight 50 --min-size 25 --checkname v5000_result --ar 1 --SAVE-MODEL

Applications

Coming soon!

Official Pytorch implementation of Meta Internal Learning

Related tags

Overview

Meta Internal Learning

Requirements

Training on a dataset

Examples from paper -

Applications

Owner

A library for hidden semi-Markov models with explicit durations

Anime Face Detector using mmdet and mmpose

Code for "Reconstructing 3D Human Pose by Watching Humans in the Mirror", CVPR 2021 oral

SeqAttack: a framework for adversarial attacks on token classification models

PyTorchVideo is a deeplearning library with a focus on video understanding work

(AAAI 2021) Progressive One-shot Human Parsing

Code for our WACV 2022 paper "Hyper-Convolution Networks for Biomedical Image Segmentation"

Animate molecular orbital transitions using Psi4 and Blender

Coursera - Quiz & Assignment of Coursera

[AI6101] Introduction to AI & AI Ethics is a core course of MSAI, SCSE, NTU, Singapore

Codes for our paper The Stem Cell Hypothesis: Dilemma behind Multi-Task Learning with Transformer Encoders published to EMNLP 2021.

An implementation of the Contrast Predictive Coding (CPC) method to train audio features in an unsupervised fashion.

PyTorch implementation of Neural Dual Contouring.

SSD-based Object Detection in PyTorch

Code to accompany our paper "Continual Learning Through Synaptic Intelligence" ICML 2017

A lightweight deep network for fast and accurate optical flow estimation.

Codes for [NeurIPS'21] You are caught stealing my winning lottery ticket! Making a lottery ticket claim its ownership.

A configurable, tunable, and reproducible library for CTR prediction

Modular Probabilistic Programming on MXNet

Weakly Supervised 3D Object Detection from Point Cloud with Only Image Level Annotation