Python model_classification_testの例

コード例 #1

ファイル: solve.py プロジェクト: soltrinox/romans_stack

#!/usr/bin/python
# -*- coding: utf-8 -*-
# @Author: Kevin Sun
# @Date:   2016-11-03 18:35:52
# @Last Modified by:   Kevin Sun
# @Last Modified time: 2016-11-06 05:34:28


import caffe
import score
import numpy as np


weights = '/home/kevin/models/shared_net/rgb_pretrained_model.caffemodel'

# init
caffe.set_device(0)
caffe.set_mode_gpu()

solver = caffe.SGDSolver('solver.prototxt')
# solver.net.copy_from(weights)

test_list = np.loadtxt('/home/kevin/dataset/processed_data3/test_object.txt', dtype=str)
# test_list = test_list[::10,:]

for i in range(30):
	solver.step(1000)
	score.model_classification_test(solver, '/home/kevin/tmp/'.format(i), test_list, layer='depth_fc8', gt='label')

コード例 #2

ファイル: solve.py プロジェクト: soltrinox/romans_stack

# -*- coding: utf-8 -*-
# @Author: Kevin Sun
# @Date:   2016-11-03 18:35:52
# @Last Modified by:   Kevin Sun
# @Last Modified time: 2016-11-06 05:34:28

import caffe
import score
import numpy as np

# weights = './vgg_fine_tuning_full_conv.caffemodel'

# init
caffe.set_device(0)
caffe.set_mode_gpu()

solver = caffe.SGDSolver('solver.prototxt')
# solver.net.copy_from(weights)

test_list = np.loadtxt('/home/kevin/dataset/processed_data/test.txt',
                       dtype=str)
# test_list = test_list[::10,:]

for i in range(15):
    solver.step(1000)
    score.model_classification_test(solver,
                                    './tmp/'.format(i),
                                    test_list,
                                    layer='fc8',
                                    gt='label')