Skip to content

Instantly share code, notes, and snippets.

View finlay-liu's full-sized avatar

Yuzhong Liu finlay-liu

View GitHub Profile
@finlay-liu
finlay-liu / tf_keras.py
Last active March 1, 2017 00:51
tensorflow with keras example
import tensorflow as tf
from keras.metrics import categorical_accuracy as accuracy
sess = tf.Session()
from keras import backend as K
K.set_session(sess)
img = tf.placeholder(tf.float32, shape=(None, 784))
from keras.layers import Dense
@finlay-liu
finlay-liu / mxnet-batch-predict.py
Last active April 18, 2019 12:45
mxnet-batch-predict.py
# -*- coding: utf-8 -*-
import os, sys, codecs
import glob
import numpy as np
import cv2
import imagehash
from PIL import Image
@finlay-liu
finlay-liu / weight_init.py
Created June 20, 2019 10:07 — forked from jeasinema/weight_init.py
A simple script for parameter initialization for PyTorch
#!/usr/bin/env python
# -*- coding:UTF-8 -*-
import torch
import torch.nn as nn
import torch.nn.init as init
def weight_init(m):
'''