gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-02-22 01:45:56 +01:00
Code Issues Projects Releases Wiki Activity

Channel order transpose, for image embedder. Updated unit test. (#87)

Browse Source
This commit is contained in:
Thom Lane
2018-11-19 05:39:03 -08:00
committed by Gal Novik
parent ff816b347d
commit 7ba1a4393f
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/tests/architectures/mxnet_components/embedders/test_image_embedder.py
View File

@@ -15,7 +15,8 @@ def test_image_embedder():
params = InputEmbedderParameters(scheme=EmbedderScheme.Medium)
emb = ImageEmbedder(params=params)
emb.initialize()
input_data = mx.nd.random.uniform(low=0, high=1, shape=(10, 3, 244, 244))
# input is NHWC, and not MXNet default NCHW
input_data = mx.nd.random.uniform(low=0, high=1, shape=(10, 244, 244, 3))
output = emb(input_data)
assert len(output.shape) == 2 # since last block was flatten
assert output.shape[0] == 10 # since batch_size is 10