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

Deep learning with TensorFlow and earth observation data.

Learning Generative Models of Textured 3D Meshes from Real-World Images, ICCV 2021

PyTorch implementation of ICLR 2022 paper PiCO: Contrastive Label Disambiguation for Partial Label Learning

Pytorch implementation of the paper: "SAPNet: Segmentation-Aware Progressive Network for Perceptual Contrastive Image Deraining"

A curated list of Machine Learning and Deep Learning tutorials in Jupyter Notebook format ready to run in Google Colaboratory

AdaMML: Adaptive Multi-Modal Learning for Efficient Video Recognition

Code for WECHSEL: Effective initialization of subword embeddings for cross-lingual transfer of monolingual language models.

FG-transformer-TTS Fine-grained style control in transformer-based text-to-speech synthesis

Unsupervised Foreground Extraction via Deep Region Competition

This repository collects 100 papers related to negative sampling methods.

Official implementation of the paper "Lightweight Deep CNN for Natural Image Matting via Similarity Preserving Knowledge Distillation"

[NeurIPS'21] "AugMax: Adversarial Composition of Random Augmentations for Robust Training" by Haotao Wang, Chaowei Xiao, Jean Kossaifi, Zhiding Yu, Animashree Anandkumar, and Zhangyang Wang.

A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".

subpixel: A subpixel convnet for super resolution with Tensorflow

Code for Reciprocal Adversarial Learning for Brain Tumor Segmentation: A Solution to BraTS Challenge 2021 Segmentation Task

Pixel Consensus Voting for Panoptic Segmentation (CVPR 2020)

Code for "Solving Graph-based Public Good Games with Tree Search and Imitation Learning"

Code release for Local Light Field Fusion at SIGGRAPH 2019

TensorFlow implementation of AlexNet and its training and testing on ImageNet ILSVRC 2012 dataset

A non-linear, non-parametric Machine Learning method capable of modeling complex datasets