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

Human Pose Detection on EdgeTPU

Implementation of "Semi-supervised Domain Adaptive Structure Learning"

Yolov5 + Deep Sort with PyTorch

Codes for paper "Towards Diverse Paragraph Captioning for Untrimmed Videos". CVPR 2021

Generating images from caption and vice versa via CLIP-Guided Generative Latent Space Search

这是一个yolox-pytorch的源码，可以用于训练自己的模型。

Download files from DSpace systems (because for some reason DSpace won't let you)

The code for the NeurIPS 2021 paper "A Unified View of cGANs with and without Classifiers".

Official implementation of ACMMM'20 paper 'Self-supervised Video Representation Learning Using Inter-intra Contrastive Framework'

PyTorch implementation of Munchausen Reinforcement Learning based on DQN and SAC. Handles discrete and continuous action spaces

Generating Band-Limited Adversarial Surfaces Using Neural Networks

Partial implementation of ODE-GAN technique from the paper Training Generative Adversarial Networks by Solving Ordinary Differential Equations

Minimal fastai code needed for working with pytorch

Self Governing Neural Networks (SGNN): the Projection Layer

Tensorflow implementation of soft-attention mechanism for video caption generation.

An educational resource to help anyone learn deep reinforcement learning.

The source code of "SIDE: Center-based Stereo 3D Detector with Structure-aware Instance Depth Estimation", accepted to WACV 2022.

PyTorch Lightning + Hydra. A feature-rich template for rapid, scalable and reproducible ML experimentation with best practices. ⚡🔥⚡

Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR 2018 (spotlight)

SeisComP/SeisBench interface to enable deep-learning (re)picking in SeisComP