Skip to content

Instantly share code, notes, and snippets.

View SeaOfOcean's full-sized avatar

Xianyan Jia SeaOfOcean

69 followers · 17 following
  • Alibaba Cloud
View GitHub Profile
@SeaOfOcean
SeaOfOcean / pvanet.prototxt
Created October 9, 2016 02:29
name: "PVANET"
################################################################################
## Input
################################################################################
input: "data"
input_shape {
dim: 1
dim: 3
@SeaOfOcean
SeaOfOcean / finetune.pt
Last active October 10, 2016 11:03
name: "PVANET finetune"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
@SeaOfOcean
SeaOfOcean / train.pt
Last active October 10, 2016 11:03
name: "PVANet train"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
@SeaOfOcean
SeaOfOcean / test.pt
Last active October 10, 2016 11:03
name: "PVANET test"
################################################################################
## Input
################################################################################
input: "data"
input_shape {
dim: 1
dim: 3
@SeaOfOcean
SeaOfOcean / vgg_faster_stage1.pt
Created November 7, 2016 01:41
name: "VGG_ILSVRC_16_layers"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
@SeaOfOcean
SeaOfOcean / vgg16 train
Created February 21, 2017 02:01
name: "VGG_ILSVRC_16_layers"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
@SeaOfOcean
SeaOfOcean / install.sh
Last active September 28, 2017 01:38
1. clone analytics-zoo project
```bash
git clone https://github.com/intel-analytics/analytics-zoo.git
```
2. install image transformer library
```
# mvn install image transformer lib
cd transform/vision
mvn clean install
@SeaOfOcean
SeaOfOcean / init_engine.py
Created September 28, 2017 01:55
from bigdl.util.common import *
init_engine()
@SeaOfOcean
SeaOfOcean / create_image_rdd.py
Created September 28, 2017 01:59
import cv2
folder = "images"
imageFiles = os.listdir(folder)
images = []
for f in imageFiles:
image = cv2.imread(folder + '/' + f)
images.append(image)
img_rdd = sc.parallelize(images)
@SeaOfOcean
SeaOfOcean / load_model.py
Created September 28, 2017 02:02
from bigdl.nn.layer import Model
# load model
model = Model.load("bigdl_mobilenet.model")
print 'load model done'