Skip to content

Instantly share code, notes, and snippets.

View hiwonjoon's full-sized avatar
🎓

Wonjoon Goo hiwonjoon

🎓
View GitHub Profile
@hiwonjoon
hiwonjoon / converter
Last active February 14, 2017 10:59
Cifar 10 data set converter from pylearn2 format to level db
import caffe_pb2 as proto
import leveldb
import sys
db = leveldb.LevelDB('./cifar_test',block_size=40 * (2 << 20) )
import struct
def unpickle(file):
import cPickle
@hiwonjoon
hiwonjoon / vgg_extract.py
Created January 27, 2017 17:35
Feature Extrcation and Write TF Record example
import numpy as np
import os
import tensorflow as tf
import vgg
import vgg_preprocessing
from pycocotools.coco import COCO
slim = tf.contrib.slim
LOG_DIR = './log/fe'
@hiwonjoon
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',
@hiwonjoon
hiwonjoon / q_learning_v0.py
Created June 19, 2017 03:54
Deep Q learning algorithm for OpenAI gym CartPole-v0 env.
import numpy as np
import sys
import random
import tensorflow as tf
import itertools
slim = tf.contrib.slim
BATCH_SIZE = 32
TRAIN_ITER = 1000
EARLY_TRAIN_TERMINATE_CRITERIA = 1e-4;
@hiwonjoon
hiwonjoon / q_learning_v1.py
Created June 19, 2017 04:26
Q-learning algorithm for OpenAI Gym CartPole-v1
import numpy as np
import sys
import random
import tensorflow as tf
import itertools
slim = tf.contrib.slim
BATCH_SIZE = 32
TRAIN_ITER = 1000
EARLY_TRAIN_TERMINATE_CRITERIA = 1e-3;
@hiwonjoon
hiwonjoon / tf-a3c-gpu.md
Created July 4, 2017 16:15
Tensorflow implementation of A3C algorithm using GPU

If you want to know how to reproduce this results, come to visit my github repo! 🐶

@hiwonjoon
hiwonjoon / ubuntu16_tensorflow_cuda8.sh
Created August 17, 2017 20:37 — forked from ksopyla/ubuntu16_tensorflow_cuda8.sh
How to set up tensorflow with CUDA 8 cuDNN 5.1 in virtualenv with Python 3.5 on Ubuntu 16.04 http://ksopyla.com/2017/02/tensorflow-gpu-virtualenv-python3/
# This is shorthened version of blog post
# http://ksopyla.com/2017/02/tensorflow-gpu-virtualenv-python3/
# update packages
sudo apt-get update
sudo apt-get upgrade
#Add the ppa repo for NVIDIA graphics driver
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
import better_exceptions
import os
import numpy as np
from tqdm import tqdm
from keras.models import model_from_json
import c3d_model
import sys
import keras.backend as K
dim_ordering = K.image_dim_ordering()
@hiwonjoon
hiwonjoon / build_local_tmux.sh
Last active April 30, 2020 20:03 — forked from joegross/build_local_tmux.sh
Install tmux without root
#!/bin/bash
set -e
TMUX_VERSION=2.4
LIBEVENT_VERSION=2.0.22-stable
NCURSES_VERSION=6.0
mkdir -p $HOME/.local
mkdir -p $HOME/src
@hiwonjoon
hiwonjoon / Frames-to-GIF.py
Last active August 4, 2019 16:31
Code Snippets for displaying image frames (NWHC format, numpy array) in Jupyter Notebook
import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation, rc
from IPython.core.display import display, HTML
%matplotlib inline
%load_ext autoreload
%autoreload 2
def display_frames_as_gif(frames):