Home

PyTorch Code for vid2vid

Info Title: Video-to-Video Synthesis Project YouTube(short) YouTube(full) Paper Note Prerequisites Linux or macOS Python 3 NVIDIA GPU + CUDA cuDNN PyTorch 0.4 Getting Started Installation Install python libraries dominate and requests. pip install dominate requests If ...

Read more

PyTorch Code for pix2pixHD

Info Title: High-Resolution Image Synthesis and Semantic Manipulation with Conditional GANs PyTorch Code Project Youtube Paper Note Prerequisites Linux or macOS Python 2 or 3 NVIDIA GPU (11G memory or larger) + CUDA cuDNN Getting Started Installation Install PyTorch and dependencies ...

Read more

PyTorch Code for BicycleGAN

Info Title: Toward multimodal image-to-image translation PyTorch Code Project Page Paper Video Note Prerequisites Linux or macOS Python 3 CPU or NVIDIA GPU + CUDA CuDNN Getting Started Installation Clone this repo: git clone -b master --single-branch https://github.com/junyanz/Bi...

Read more

THOP: Python Tool for Counting MACs/FLOPs in PyTorch

Source Code on GitHub How to install pip install thop (now continously intergrated on Github actions) OR pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git How to use Basic usage from torchvision.models import resnet50 from thop import profile model = resnet50() input = torch.randn(1, 3, 224, 224) flop...

Read more

argparse-based Python Option Configuration Codebase

BaseOptions Class import argparse import os from util import util import torch import models import data class BaseOptions(): """This class defines options used during both training and test time. It also implements several helper functions such as parsing, printing, and saving the options. It also gathers additional options defi...

Read more

PyTorch Code for CycleGAN

Info Title: Image-to-Image Translation with Conditional Adversarial Networks PyTorch Code Project Paper Torch Note Prerequisites Linux or macOS Python 3 CPU or NVIDIA GPU + CUDA CuDNN Getting Started Installation Clone this repo: git clone https://github.com/junyanz/pytorch-Cycl...

Read more