Skip to content

Instantly share code, notes, and snippets.

View ProGamerGov's full-sized avatar

ProGamerGov

128 followers · 1 following
  • Multiverse
View GitHub Profile
@ProGamerGov
ProGamerGov / neural_style_test_loss.py
Last active December 8, 2019 00:36
import os
import copy
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision.transforms as transforms
from PIL import Image
from CaffeLoader import loadCaffemodel, ModelParallel
@ProGamerGov
ProGamerGov / download_gdrive_models.sh
Last active November 30, 2019 17:11
#!/bin/bash
fileid="1qWjJd-mVNxvHhzc9XsU6LjC0pQXIwW4t"
filename="models/VGG16-Stylized-ImageNet.pth"
curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=${fileid}" > /dev/null
curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=${fileid}" -o ${filename}
fileid="1EU-F9ugeIeTO9ay4PinzsBXgEuCYBu0Z"
filename="models/VGG16_SOD_finetune.pth"
curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=${fileid}" > /dev/null
curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=${fileid}" -o ${filename}
@ProGamerGov
ProGamerGov / convert_model.py
Last active November 17, 2019 18:18
Make a model's weight and bias names be compatible with neural-style-pt
import torch
from collections import OrderedDict
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("-model_file", type=str, default='')
parser.add_argument("-output_name", type=str, default='')
params = parser.parse_args()
@ProGamerGov
ProGamerGov / notebook.ipynb
Last active November 16, 2019 02:06
neural-style-pt
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@ProGamerGov
ProGamerGov / neural_style_deepdream.py
Last active November 3, 2019 15:02
import os
import copy
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision.transforms as transforms
from PIL import Image
from CaffeLoader import loadCaffemodel, ModelParallel
@ProGamerGov
ProGamerGov / neural_style_time.py
Last active November 1, 2019 15:22
import os
import copy
import time
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision.transforms as transforms
from PIL import Image
from CaffeLoader import loadCaffemodel, ModelParallel
@ProGamerGov
ProGamerGov / neural_style_c.py
Last active November 1, 2019 15:09
import os
import copy
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision.transforms as transforms
from PIL import Image
from CaffeLoader import loadCaffemodel, ModelParallel
@ProGamerGov
ProGamerGov / CaffeLoaderMod.py
Last active November 1, 2019 14:58
import torch
import torch.nn as nn
class VGG(nn.Module):
def __init__(self, features, num_classes=1000):
super(VGG, self).__init__()
self.features = features
self.classifier = nn.Sequential(
nn.Linear(512 * 7 * 7, 4096),
@ProGamerGov
ProGamerGov / ModelParallel.py
Last active October 15, 2019 19:16
# Inspired by: https://github.com/torch/nn/blob/master/GPU.lua
# And: https://github.com/jcjohnson/neural-style/blob/master/neural_style.lua#L360
# As seen in: https://github.com/ProGamerGov/neural-style-pt
import torch
import torch.nn as nn
class ModelParallel(nn.Module):
r"""Splits a sequential network across multiple devices.
@ProGamerGov
ProGamerGov / README.md
Last active October 6, 2018 18:56

Multiscale Resolution Scripts

Setup

First download the linear-color-transfer.py, and lum-transfer.py scripts from Neural-Tools:

wget -c https://raw.githubusercontent.com/ProGamerGov/Neural-Tools/master/linear-color-transfer.py

wget -c https://raw.githubusercontent.com/ProGamerGov/Neural-Tools/master/lum-transfer.py