Skip to content

Commit db30250

Browse files
committed
README tweak
1 parent b890a4e commit db30250

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

README.md

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,15 @@ model:add(nn.SoftMax())
137137
model:evaluate()
138138
model:cuda()
139139

140+
-- prepare detector
141+
image_transformer= nnf.ImageTransformer{mean_pix={102.9801,115.9465,122.7717},
142+
raw_scale = 255,
143+
swap = {3,2,1}}
144+
feat_provider = nnf.FRCNN{image_transformer=image_transformer}
145+
feat_provider:evaluate() -- testing mode
146+
147+
detector = nnf.ImageDetect(model, feat_provider)
148+
140149
-- Load an image
141150
I = image.lena()
142151
-- generate some random bounding boxes
@@ -147,12 +156,7 @@ bboxes:select(2,2):random(1,I:size(2)/2)
147156
bboxes:select(2,3):random(I:size(3)/2+1,I:size(3))
148157
bboxes:select(2,4):random(I:size(2)/2+1,I:size(2))
149158

150-
image_transformer= nnf.ImageTransformer{mean_pix={102.9801,115.9465,122.7717},
151-
raw_scale = 255,
152-
swap = {3,2,1}}
153-
feat_provider = nnf.FRCNN{image_transformer=image_transformer}
154-
155-
detector = nnf.ImageDetect(model, feat_provider)
159+
-- detect !
156160
scores, bboxes = detector:detect(I, bboxes)
157161

158162
-- visualization

0 commit comments

Comments
 (0)