Class: DSelect

Inherits:
Object
  • Object
show all
Defined in:
lib/deepselect.rb

Constant Summary collapse

@@vgg16_obj = 
nil

Class Method Summary collapse

Class Method Details

.compare(f1, f2) ⇒ Object

image同士を比較して特徴量出す



15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/deepselect.rb', line 15

def self.compare(f1, f2)
	# 引数がRmagick imageならfeature vectorへ変換
	if f1.class == Magick::Image and f2.class == Magick::Image
		f1,f2 = f1.to_vector, f2.to_vector # さすがに他のライブラリと被りそうな気が
	end

	# cos similarity
	v1 = Vector.elements(f1)
	v2 = Vector.elements(f2)
	return v2.inner_product(v1)/(v1.norm() * v2.norm())
end

.id ⇒ Object

…？ modelの各layerのid



53
54
55
56
57
58
 
# File 'lib/deepselect.rb', line 53

def self.id
	conv1_id = '140326425860192'.freeze
	fc6_id = '140326200777584'.freeze
	softmax_id = '140326200803680'.freeze
	{conv1_id: conv1_id, fc6_id: fc6_id, softmax_id:softmax_id}
end

.model_opt ⇒ Object

モデルの詳細



61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
 
# File 'lib/deepselect.rb', line 61

def self.model_opt
	conv1_id, fc6_id, softmax_id = self.id.values

	input_shape = {
		channel_num: 3,
 			width: 224,
 			height: 224
	}
	model_opt = {
		backend: 'mkldnn',
		input_layers: [
			{
				name: conv1_id,
				dims: [
					1, #image_list.size, # batch size
					input_shape[:channel_num],
					input_shape[:height],
					input_shape[:width],
				]
			}
		],
		output_layers: [fc6_id, softmax_id]
	}
end

.vgg16 ⇒ Object

vgg16返す



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/deepselect.rb', line 28

def self.vgg16
	# 読み込み済みなら読み込んであるモデルを返す
	return @@vgg16_obj if @@vgg16_obj

	vgg16_dir = File.expand_path('./deepselect/data/VGG16.onnx', __dir__)
	if File.exist?(vgg16_dir)
		@@vgg16_obj = Menoh::Menoh.new(vgg16_dir)
	else
		# menohと同じようにモデルをダウンロードする
		# url勝手に使うのダメだったら教えてください...！
		model_parent_dir = File.expand_path('./deepselect/data', __dir__)
		FileUtils.mkdir(model_parent_dir) unless File.exist?(model_parent_dir)
		puts "model data downloading... (first time only)"
		url = 'https://www.dropbox.com/s/bjfn9kehukpbmcm/VGG16.onnx?dl=1'
		open(url) do |file|
			File.open(vgg16_dir, "wb") do |out|
				out.write(file.read)
			end
		end
		@@vgg16_obj = Menoh::Menoh.new(vgg16_dir)
	end
	return  @@vgg16_obj
end