Semantic-aware Grad-GAN for Virtual-to-Real Urban Scene Adaption

Last update: Jun 07, 2022

Related tags

Deep Learning SG-GAN

Overview

SG-GAN

TensorFlow implementation of SG-GAN.

Prerequisites

TensorFlow (implemented in v1.3)
numpy
scipy
pillow

Getting Started

Train

Prepare dataset. We present example images in datasets folder for reference as data format, and scripts prepare_data.py and segment_class.py for reference in preparing dataset.
Train a model:

CUDA_VISIBLE_DEVICES=0 python main.py

Models are saved to ./checkpoints/ (can be changed by passing --checkpoint_dir=your_dir).

Continue training a model (useful for updating parameters)

CUDA_VISIBLE_DEVICES=0 python main.py --continue_train 1

Test

Finally, test the model:

CUDA_VISIBLE_DEVICES=0 python main.py --phase test --img_width 2048 --img_height 1024

Adapted test images will be outputted to ./test/ (can be changed by passing --test_dir=your_dir).

Reference

The TensorFlow implementation of CycleGAN, https://github.com/xhujoy/CycleGAN-tensorflow

Owner

lplcor

Machine Learning Artist 👩‍🎨 (CV, DL, AI)

GitHub Repository https://arxiv.org/abs/1801.01726

Joint deep network for feature line detection and description

SOLD² - Self-supervised Occlusion-aware Line Description and Detection This repository contains the implementation of the paper: SOLD² : Self-supervis

427 Dec 27, 2022

Learning based AI for playing multi-round Koi-Koi hanafuda card games. Have fun.

Koi-Koi AI Learning based AI for playing multi-round Koi-Koi hanafuda card games. Platform Python PyTorch PySimpleGUI (for the interface playing vs AI

10 Nov 20, 2022

[Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021

Convolutional MLP ConvMLP: Hierarchical Convolutional MLPs for Vision Preprint link: ConvMLP: Hierarchical Convolutional MLPs for Vision By Jiachen Li

143 Jan 03, 2023

End-to-end speech secognition toolkit

End-to-end speech secognition toolkit This is an E2E ASR toolkit modified from Espnet1 (version 0.9.9). This is the official implementation of paper:

147 Dec 28, 2022

Multi-Objective Loss Balancing for Physics-Informed Deep Learning

Multi-Objective Loss Balancing for Physics-Informed Deep Learning Code for ReLoBRaLo. Abstract Physics Informed Neural Networks (PINN) are algorithms

16 Dec 12, 2022

A Light CNN for Deep Face Representation with Noisy Labels

A Light CNN for Deep Face Representation with Noisy Labels Citation If you use our models, please cite the following paper: @article{wulight, title=

715 Nov 05, 2022

GNPy: Optical Route Planning and DWDM Network Optimization

GNPy is an open-source, community-developed library for building route planning and optimization tools in real-world mesh optical networks

140 Dec 19, 2022

Code for the paper "Implicit Representations of Meaning in Neural Language Models"

Implicit Representations of Meaning in Neural Language Models Preliminaries Create and set up a conda environment as follows: conda create -n state-pr

39 Nov 03, 2022

The official codes for the ICCV2021 presentation "Uniformity in Heterogeneity: Diving Deep into Count Interval Partition for Crowd Counting"

UEPNet (ICCV2021 Poster Presentation) This repository contains codes for the official implementation in PyTorch of UEPNet as described in Uniformity i

15 Dec 14, 2022

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

Unofficial PyTorch implementation of "Zero-Shot" Super-Resolution using Deep Internal Learning Unofficial Implementation of 1712.06087 "Zero-Shot" Sup

196 Nov 27, 2022

Harmonic Memory Networks for Graph Completion

HMemNetworks Code and documentation for Harmonic Memory Networks, a series of models for compositionally assembling representations of graph elements

0 Oct 27, 2021

Using Random Effects to Account for High-Cardinality Categorical Features and Repeated Measures in Deep Neural Networks

LMMNN Using Random Effects to Account for High-Cardinality Categorical Features and Repeated Measures in Deep Neural Networks This is the working dire

10 Nov 02, 2022

李云龙二次元风格化!打滚卖萌，使用了animeGANv2进行了视频的风格迁移

李云龙二次元风格化！一键star、fork，你也可以生成这样的团长！打滚卖萌求star求fork! 0.效果展示视频效果前往B站观看效果最佳：李云龙二次元风格化： github开源repo：李云龙二次元风格化百度AIstudio开源地址,一键fork即可运行: 李云龙二次元风格化！一键fork

44 Dec 04, 2022

PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)

mlp-mixer-pytorch PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021) Usage import torch from mlp_mixer

27 Jul 09, 2022

Training BERT with Compute/Time (Academic) Budget

Training BERT with Compute/Time (Academic) Budget This repository contains scripts for pre-training and finetuning BERT-like models with limited time

263 Jan 07, 2023

Conjugated Discrete Distributions for Distributional Reinforcement Learning (C2D)

Conjugated Discrete Distributions for Distributional Reinforcement Learning (C2D) Code & Data Appendix for Conjugated Discrete Distributions for Distr

1 Jan 11, 2022

HAR-stacked-residual-bidir-LSTMs - Deep stacked residual bidirectional LSTMs for HAR

HAR-stacked-residual-bidir-LSTM The project is based on this repository which is presented as a tutorial. It consists of Human Activity Recognition (H

287 Dec 27, 2022

Calling Julia from Python - an experiment on data loading

Calling Julia from Python - an experiment on data loading See the slides. TLDR After reading Patrick's blog post, we decided to try to replace C++ wit

8 Jun 07, 2022

A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks

Simple implementation of Equivariant GNN A short implementation of E(n) Equivariant Graph Neural Networks for HOMO energy prediction. Just 50 lines of

97 Dec 23, 2022

Code for the Paper: Alexandra Lindt and Emiel Hoogeboom.

Discrete Denoising Flows This repository contains the code for the experiments presented in the paper Discrete Denoising Flows [1]. To give a short ov

3 Oct 09, 2022