gluon-cv icon indicating copy to clipboard operation
gluon-cv copied to clipboard

Published 20 hours ago • verified publisher icon dmlc

fix segmentation testing and support CPU

Open wkcn opened this issue 5 years ago • 5 comments

Hi, there.

In the testing script of segmentation, the shape of predict[0] is (num_cls, height, weight).

In the line 71 of scripts/segmentation/test.py, the axis of mx.nd.argmax should be 0.

Besides, I modify the code to support the inference of segmentation model on CPU.

Thank you!

wkcn avatar Jul 29 '19 10:07 wkcn

Job PR-888-1 is done. Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-888/1/index.html Code coverage of this PR: pr.svg vs. Master: master.svg

mli avatar Jul 29 '19 11:07 mli

Job PR-888-2 is done. Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-888/2/index.html Code coverage of this PR: pr.svg vs. Master: master.svg

mli avatar Jul 29 '19 11:07 mli

Job PR-888-3 is done. Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-888/3/index.html Code coverage of this PR: pr.svg vs. Master: master.svg

mli avatar Jul 30 '19 01:07 mli

Please rebase the code to resolve conflicts since this PR has been merged into master.

wuxun-zhang avatar Aug 20 '19 06:08 wuxun-zhang

Job PR-888-4 is done. Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-888/4/index.html Code coverage of this PR: pr.svg vs. Master: master.svg

mli avatar Aug 20 '19 09:08 mli