zijunwei zijunwei

Research Scientist @ Adobe Photoshop

mkocabas / coco.sh

Created April 9, 2018 09:41

Download COCO dataset. Run under 'datasets' directory.

	mkdir coco
	cd coco
	mkdir images
	cd images

	wget http://images.cocodataset.org/zips/train2017.zip
	wget http://images.cocodataset.org/zips/val2017.zip
	wget http://images.cocodataset.org/zips/test2017.zip
	wget http://images.cocodataset.org/zips/unlabeled2017.zip

apaszke / graph_viz.py

Last active April 7, 2022 02:46

	from graphviz import Digraph
	import torch
	from torch.autograd import Variable, Function

	def iter_graph(root, callback):
	queue = [root]
	seen = set()
	while queue:
	fn = queue.pop()
	if fn in seen:

GavinXing / mini-word2vec-pytorch.py

Created August 6, 2017 01:44

A complete word2vec based on pytorch tutorial

	# encoding=utf-8
	# Project: learn-pytorch
	# Author: xingjunjie github: @gavinxing
	# Create Time: 29/07/2017 11:58 AM on PyCharm
	# Basic template from http://pytorch.org/tutorials/beginner/nlp/word_embeddings_tutorial.html

	import torch
	import torch.nn as nn
	import torch.autograd as autograd
	import torch.optim as optim

MatthieuBizien / fast_glove.py

Created June 7, 2017 14:53

	# coding: utf-8

	import logging
	import re
	from collections import Counter

	import numpy as np
	import torch
	from sklearn.datasets import fetch_20newsgroups
	from torch.autograd import Variable

apaszke / bad_grad_viz.py

Last active February 13, 2026 08:57

	from graphviz import Digraph
	import torch
	from torch.autograd import Variable, Function

	def iter_graph(root, callback):
	queue = [root]
	seen = set()
	while queue:
	fn = queue.pop()
	if fn in seen:

hiwonjoon / python-ffmpeg.py

Last active November 26, 2023 16:11

ffmpeg and ffprobe subprocess call in python; extract specific frame at some timepoint, extract duration of a video

	import subprocess
	import datetime
	import numpy as np

	THREAD_NUM=4

	def get_video_info(fileloc) :
	command = ['ffprobe',
	'-v', 'fatal',
	'-show_entries', 'stream=width,height,r_frame_rate,duration',

oeway / imageUtils.py

Last active May 8, 2024 14:21

Improved image transform functions for dense predictions (for pytorch, keras etc.)

	import numpy as np
	import scipy
	import scipy.ndimage
	from scipy.ndimage.filters import gaussian_filter
	from scipy.ndimage.interpolation import map_coordinates
	import collections
	from PIL import Image
	import numbers

	__author__ = "Wei OUYANG"

mjdietzx / waya-dl-setup.sh

Last active September 20, 2025 11:52

Install CUDA Toolkit v8.0 and cuDNN v6.0 on Ubuntu 16.04

	#!/bin/bash

	# install CUDA Toolkit v8.0
	# instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb (network))
	CUDA_REPO_PKG="cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
	wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/${CUDA_REPO_PKG}
	sudo dpkg -i ${CUDA_REPO_PKG}
	sudo apt-get update
	sudo apt-get -y install cuda

kenzotakahashi / pytorch_gru.py

Last active February 4, 2020 17:35

PyTorch GRU example with a Keras-like interface.

	import numpy as np
	from sklearn.model_selection import train_test_split

	import torch
	import torch.nn as nn
	from torch.autograd import Variable

	np.random.seed(1337)

	MAX_LEN = 30

xmfbit / pytorch_mnist.py

Last active March 4, 2023 19:45

an example of pytorch on mnist dataset

	import os
	import torch
	import torch.nn as nn
	from torch.autograd import Variable
	import torchvision.datasets as dset
	import torchvision.transforms as transforms
	import torch.nn.functional as F
	import torch.optim as optim
	## load mnist dataset
	use_cuda = torch.cuda.is_available()