mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2024-11-28 13:10:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #17469 from l-bat:fix_virtual_try_on

Browse Source
This commit is contained in:
Alexander Alekhin 2020-06-04 10:32:30 +00:00
commit 08d1c54364
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/dnn/src/onnx/onnx_importer.cpp
View File

@ -806,6 +806,10 @@ void ONNXImporter::populateNet(Net dstNet)
{ {
layerParams.type = "ELU"; layerParams.type = "ELU";
} }
else if (layer_type == "Tanh")
{
layerParams.type = "TanH";
}
else if (layer_type == "PRelu") else if (layer_type == "PRelu")
{ {
layerParams.type = "PReLU"; layerParams.type = "PReLU";

2
samples/dnn/virtual_try_on.py
View File

@ -185,7 +185,7 @@ class CpVton(object):
agnostic = np.concatenate((res_shape, img_head, pose_map), axis=0) agnostic = np.concatenate((res_shape, img_head, pose_map), axis=0)
agnostic = np.expand_dims(agnostic, axis=0) agnostic = np.expand_dims(agnostic, axis=0)
return agnostic return agnostic.astype(np.float32)
def get_warped_cloth(self, cloth_img, agnostic, height=256, width=192): def get_warped_cloth(self, cloth_img, agnostic, height=256, width=192):
cloth = cv.dnn.blobFromImage(cloth_img, 1.0 / 127.5, (width, height), mean=(127.5, 127.5, 127.5), swapRB=True) cloth = cv.dnn.blobFromImage(cloth_img, 1.0 / 127.5, (width, height), mean=(127.5, 127.5, 127.5), swapRB=True)