Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

yolov3_resnet18 bugfix

This commit is contained in:
zhangxiaoxiao 2021-06-21 15:50:37 +08:00
parent a8553078b9
commit bc8dddf682
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/cv/yolov3_resnet18/ascend310_infer/src/main.cc
View File

@ -74,7 +74,7 @@ int PadImage(const MSTensor &input, MSTensor *output) {
widthSize = shape[0]*heightScale; widthSize = shape[0]*heightScale;
heightSize = shape[1]*heightScale; heightSize = shape[1]*heightScale;
} }
std::shared_ptr<TensorTransform> resize(new Resize({widthSize, heightSize}, InterpolationMode::kArea)); std::shared_ptr<TensorTransform> resize(new Resize({widthSize, heightSize}, InterpolationMode::kCubicPil));
Execute composeResize({resize}); Execute composeResize({resize});
Status ret = composeResize(input, &imgResize); Status ret = composeResize(input, &imgResize);
if (ret != kSuccess) { if (ret != kSuccess) {
@ -110,6 +110,7 @@ int main(int argc, char **argv) {
auto context = std::make_shared<Context>(); auto context = std::make_shared<Context>();
auto ascend310 = std::make_shared<mindspore::Ascend310DeviceInfo>(); auto ascend310 = std::make_shared<mindspore::Ascend310DeviceInfo>();
ascend310->SetDeviceID(FLAGS_device_id); ascend310->SetDeviceID(FLAGS_device_id);
ascend310->SetBufferOptimizeMode("off_optimize");
context->MutableDeviceInfo().push_back(ascend310); context->MutableDeviceInfo().push_back(ascend310);
mindspore::Graph graph; mindspore::Graph graph;
Serialization::Load(FLAGS_mindir_path, ModelType::kMindIR, &graph); Serialization::Load(FLAGS_mindir_path, ModelType::kMindIR, &graph);