[jacinto-ai/pytorch-jacinto-ai-devkit.git] / modules / pytorch_jacinto_ai / xvision / ops / poolers.py
diff --git a/modules/pytorch_jacinto_ai/vision/ops/poolers.py b/modules/pytorch_jacinto_ai/xvision/ops/poolers.py
similarity index 98%
rename from modules/pytorch_jacinto_ai/vision/ops/poolers.py
rename to modules/pytorch_jacinto_ai/xvision/ops/poolers.py
index d20f8a4327eb673dfaa4c9096e47fb80f8b4204a..13d1a7806320fff4a1d360d77a95d2c1d850162c 100644 (file)
rename from modules/pytorch_jacinto_ai/vision/ops/poolers.py
rename to modules/pytorch_jacinto_ai/xvision/ops/poolers.py
index d20f8a4327eb673dfaa4c9096e47fb80f8b4204a..13d1a7806320fff4a1d360d77a95d2c1d850162c 100644 (file)
Examples::
- >>> m = pytorch_jacinto_ai.vision.ops.MultiScaleRoIAlign(['feat1', 'feat3'], 3, 2)
+ >>> m = pytorch_jacinto_ai.xvision.ops.MultiScaleRoIAlign(['feat1', 'feat3'], 3, 2)
>>> i = OrderedDict()
>>> i['feat1'] = torch.rand(1, 5, 64, 64)
>>> i['feat2'] = torch.rand(1, 5, 32, 32) # this feature won't be used in the pooling