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

Repository of continual learning papers

KITTI-360 Annotation Tool is a framework that developed based on python(cherrypy + jinja2 + sqlite3) as the server end and javascript + WebGL as the front end.

Robust fine-tuning of zero-shot models

NEG loss implemented in pytorch

Implementation for our ICCV2021 paper: Internal Video Inpainting by Implicit Long-range Propagation

Vision Transformer and MLP-Mixer Architectures

Process text, including tokenizing and representing sentences as vectors and Applying some concepts like RNN, LSTM and GRU to create a classifier can detect the language in which a sentence is written from among 17 languages.

Online Pseudo Label Generation by Hierarchical Cluster Dynamics for Adaptive Person Re-identification

Session-based Recommendation, CoHHN, price preferences, interest preferences, Heterogeneous Hypergraph, Co-guided Learning, SIGIR2022

Mall-Customers-Segmentation - Customer Segmentation Using K-Means Clustering

FAST Aiming at the problems of cumbersome steps and slow download speed of GNSS data

rastrainer is a QGIS plugin to training remote sensing semantic segmentation model based on PaddlePaddle.

An end-to-end library for editing and rendering motion of 3D characters with deep learning [SIGGRAPH 2020]

Code release for ICCV 2021 paper "Anticipative Video Transformer"

PyGAD, a Python 3 library for building the genetic algorithm and training machine learning algorithms (Keras & PyTorch).

An Ensemble of CNN (Python 3.5.1 Tensorflow 1.3 numpy 1.13)

Parallel and High-Fidelity Text-to-Lip Generation; AAAI 2022 ; Official code

Official Code Release for "TIP-Adapter: Training-free clIP-Adapter for Better Vision-Language Modeling"

This is a official repository of SimViT.

Code of 3D Shape Variational Autoencoder Latent Disentanglement via Mini-Batch Feature Swapping for Bodies and Faces