cherry-pick 1.serving bug; 2.feature_extraction.md
parent
1201af5d19
commit
689f57d444
|
@ -64,12 +64,12 @@ class DetOp(Op):
|
||||||
im_scale_y, im_scale_x = self.generate_scale(raw_im)
|
im_scale_y, im_scale_x = self.generate_scale(raw_im)
|
||||||
im = self.img_preprocess(raw_im)
|
im = self.img_preprocess(raw_im)
|
||||||
|
|
||||||
|
im_shape = np.array(im.shape[1:]).reshape(-1)
|
||||||
|
scale_factor = np.array([im_scale_y, im_scale_x]).reshape(-1)
|
||||||
imgs.append({
|
imgs.append({
|
||||||
"image": im[np.newaxis, :],
|
"image": im[np.newaxis, :],
|
||||||
"im_shape":
|
"im_shape": im_shape[np.newaxis, :],
|
||||||
np.array(list(im.shape[1:])).reshape(-1)[np.newaxis, :],
|
"scale_factor": scale_factor[np.newaxis, :],
|
||||||
"scale_factor":
|
|
||||||
np.array([im_scale_y, im_scale_x]).astype('float32'),
|
|
||||||
})
|
})
|
||||||
self.raw_img = raw_imgs
|
self.raw_img = raw_imgs
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue