mirror of
https://github.com/opencv/opencv.git
synced 2026-07-30 15:53:03 +04:00
Merge pull request #21660 from TolyaTalamanov:at/handle-exception-in-streamingexecutor
[G-API] Handle exceptions in streaming executor * Handle exceptions in streaming executor * Rethrow exception in non-streaming executor * Clean up * Put more tests * Handle exceptions in IE backend * Handle exception in IE callbacks * Handle exception in GExecutor * Handle all exceptions in IE backend * Not only (std::exception& e) * Fix comments to review * Handle input exception in generic way * Fix comment * Clean up * Apply review comments * Put more comments * Fix alignment * Move test outside of HAVE_NGRAPH * Fix compilation
This commit is contained in:
Anatoliy Talamanov
committed by
GitHub
GitHub
parent
78bc11465b
commit
d98e07c3d3
@@ -2915,6 +2915,47 @@ TEST(Infer, ModelWith2DInputs)
|
||||
|
||||
#endif // HAVE_NGRAPH
|
||||
|
||||
TEST(TestAgeGender, ThrowBlobAndInputPrecisionMismatchStreaming)
|
||||
{
|
||||
const std::string device = "MYRIAD";
|
||||
skipIfDeviceNotAvailable(device);
|
||||
|
||||
initDLDTDataPath();
|
||||
|
||||
cv::gapi::ie::detail::ParamDesc params;
|
||||
// NB: Precision for inputs is U8.
|
||||
params.model_path = compileAgeGenderBlob(device);
|
||||
params.device_id = device;
|
||||
|
||||
// Configure & run G-API
|
||||
using AGInfo = std::tuple<cv::GMat, cv::GMat>;
|
||||
G_API_NET(AgeGender, <AGInfo(cv::GMat)>, "test-age-gender");
|
||||
|
||||
auto pp = cv::gapi::ie::Params<AgeGender> {
|
||||
params.model_path, params.device_id
|
||||
}.cfgOutputLayers({ "age_conv3", "prob" });
|
||||
|
||||
cv::GMat in, age, gender;
|
||||
std::tie(age, gender) = cv::gapi::infer<AgeGender>(in);
|
||||
auto pipeline = cv::GComputation(cv::GIn(in), cv::GOut(age, gender))
|
||||
.compileStreaming(cv::compile_args(cv::gapi::networks(pp)));
|
||||
|
||||
cv::Mat in_mat(320, 240, CV_32FC3);
|
||||
cv::randu(in_mat, 0, 1);
|
||||
cv::Mat gapi_age, gapi_gender;
|
||||
|
||||
pipeline.setSource(cv::gin(in_mat));
|
||||
pipeline.start();
|
||||
|
||||
// NB: Blob precision is U8, but user pass FP32 data, so exception will be thrown.
|
||||
// Now exception comes directly from IE, but since G-API has information
|
||||
// about data precision at the compile stage, consider the possibility of
|
||||
// throwing exception from there.
|
||||
for (int i = 0; i < 10; ++i) {
|
||||
EXPECT_ANY_THROW(pipeline.pull(cv::gout(gapi_age, gapi_gender)));
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace opencv_test
|
||||
|
||||
#endif // HAVE_INF_ENGINE
|
||||
|
||||
@@ -304,6 +304,66 @@ void checkPullOverload(const cv::Mat& ref,
|
||||
EXPECT_EQ(0., cv::norm(ref, out_mat, cv::NORM_INF));
|
||||
}
|
||||
|
||||
class InvalidSource : public cv::gapi::wip::IStreamSource {
|
||||
public:
|
||||
InvalidSource(const size_t throw_every_nth_frame,
|
||||
const size_t num_frames)
|
||||
: m_throw_every_nth_frame(throw_every_nth_frame),
|
||||
m_curr_frame_id(0u),
|
||||
m_num_frames(num_frames),
|
||||
m_mat(1, 1, CV_8U) {
|
||||
}
|
||||
|
||||
static std::string exception_msg()
|
||||
{
|
||||
return "InvalidSource sucessfuly failed!";
|
||||
}
|
||||
|
||||
bool pull(cv::gapi::wip::Data& d) {
|
||||
++m_curr_frame_id;
|
||||
if (m_curr_frame_id > m_num_frames) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (m_curr_frame_id % m_throw_every_nth_frame == 0) {
|
||||
throw std::logic_error(InvalidSource::exception_msg());
|
||||
return true;
|
||||
} else {
|
||||
d = cv::Mat(m_mat);
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
cv::GMetaArg descr_of() const override {
|
||||
return cv::GMetaArg{cv::descr_of(m_mat)};
|
||||
}
|
||||
|
||||
private:
|
||||
size_t m_throw_every_nth_frame;
|
||||
size_t m_curr_frame_id;
|
||||
size_t m_num_frames;
|
||||
cv::Mat m_mat;
|
||||
};
|
||||
|
||||
G_TYPED_KERNEL(GThrowExceptionOp, <GMat(GMat)>, "org.opencv.test.throw_error_op")
|
||||
{
|
||||
static GMatDesc outMeta(GMatDesc in) { return in; }
|
||||
};
|
||||
|
||||
GAPI_OCV_KERNEL(GThrowExceptionKernel, GThrowExceptionOp)
|
||||
{
|
||||
static std::string exception_msg()
|
||||
{
|
||||
return "GThrowExceptionKernel sucessfuly failed";
|
||||
}
|
||||
|
||||
static void run(const cv::Mat&, cv::Mat&)
|
||||
{
|
||||
throw std::logic_error(GThrowExceptionKernel::exception_msg());
|
||||
}
|
||||
};
|
||||
|
||||
} // anonymous namespace
|
||||
|
||||
TEST_P(GAPI_Streaming, SmokeTest_ConstInput_GMat)
|
||||
@@ -2512,5 +2572,109 @@ TEST(GAPI_Streaming, TestDesyncMediaFrameGray) {
|
||||
}
|
||||
}
|
||||
|
||||
TEST(GAPI_Streaming_Exception, SingleKernelThrow) {
|
||||
cv::GMat in;
|
||||
auto pipeline = cv::GComputation(in, GThrowExceptionOp::on(in))
|
||||
.compileStreaming(cv::compile_args(cv::gapi::kernels<GThrowExceptionKernel>()));
|
||||
|
||||
cv::Mat in_mat(cv::Size(300, 300), CV_8UC3);
|
||||
cv::randu(in_mat, cv::Scalar::all(0), cv::Scalar::all(255));
|
||||
pipeline.setSource(cv::gin(in_mat));
|
||||
pipeline.start();
|
||||
|
||||
EXPECT_THROW(
|
||||
try {
|
||||
cv::Mat out_mat;
|
||||
pipeline.pull(cv::gout(out_mat));
|
||||
} catch (const std::logic_error& e) {
|
||||
EXPECT_EQ(GThrowExceptionKernel::exception_msg(), e.what());
|
||||
throw;
|
||||
}, std::logic_error);
|
||||
}
|
||||
|
||||
TEST(GAPI_Streaming_Exception, StreamingBackendExceptionAsInput) {
|
||||
cv::GMat in;
|
||||
auto pipeline = cv::GComputation(in,
|
||||
cv::gapi::copy(GThrowExceptionOp::on(in)))
|
||||
.compileStreaming(cv::compile_args(cv::gapi::kernels<GThrowExceptionKernel>()));
|
||||
|
||||
cv::Mat in_mat(cv::Size(300, 300), CV_8UC3);
|
||||
cv::randu(in_mat, cv::Scalar::all(0), cv::Scalar::all(255));
|
||||
pipeline.setSource(cv::gin(in_mat));
|
||||
pipeline.start();
|
||||
|
||||
EXPECT_THROW(
|
||||
try {
|
||||
cv::Mat out_mat;
|
||||
pipeline.pull(cv::gout(out_mat));
|
||||
} catch (const std::logic_error& e) {
|
||||
EXPECT_EQ(GThrowExceptionKernel::exception_msg(), e.what());
|
||||
throw;
|
||||
}, std::logic_error);
|
||||
}
|
||||
|
||||
TEST(GAPI_Streaming_Exception, RegularBacckendsExceptionAsInput) {
|
||||
cv::GMat in;
|
||||
auto pipeline = cv::GComputation(in,
|
||||
cv::gapi::add(GThrowExceptionOp::on(in), GThrowExceptionOp::on(in)))
|
||||
.compileStreaming(cv::compile_args(cv::gapi::kernels<GThrowExceptionKernel>()));
|
||||
|
||||
cv::Mat in_mat(cv::Size(300, 300), CV_8UC3);
|
||||
cv::randu(in_mat, cv::Scalar::all(0), cv::Scalar::all(255));
|
||||
pipeline.setSource(cv::gin(in_mat));
|
||||
pipeline.start();
|
||||
|
||||
EXPECT_THROW(
|
||||
try {
|
||||
cv::Mat out_mat;
|
||||
pipeline.pull(cv::gout(out_mat));
|
||||
} catch (const std::logic_error& e) {
|
||||
EXPECT_EQ(GThrowExceptionKernel::exception_msg(), e.what());
|
||||
throw;
|
||||
}, std::logic_error);
|
||||
}
|
||||
|
||||
TEST(GAPI_Streaming_Exception, SourceThrow) {
|
||||
cv::GMat in;
|
||||
auto pipeline = cv::GComputation(in, cv::gapi::copy(in)).compileStreaming();
|
||||
|
||||
pipeline.setSource(std::make_shared<InvalidSource>(1u, 1u));
|
||||
pipeline.start();
|
||||
|
||||
EXPECT_THROW(
|
||||
try {
|
||||
cv::Mat out_mat;
|
||||
pipeline.pull(cv::gout(out_mat));
|
||||
} catch (const std::logic_error& e) {
|
||||
EXPECT_EQ(InvalidSource::exception_msg(), e.what());
|
||||
throw;
|
||||
}, std::logic_error);
|
||||
}
|
||||
|
||||
TEST(GAPI_Streaming_Exception, SourceThrowEverySecondFrame) {
|
||||
constexpr size_t throw_every_nth_frame = 2u;
|
||||
constexpr size_t num_frames = 10u;
|
||||
size_t curr_frame = 0;
|
||||
bool has_frame = true;
|
||||
cv::Mat out_mat;
|
||||
|
||||
cv::GMat in;
|
||||
auto pipeline = cv::GComputation(in, cv::gapi::copy(in)).compileStreaming();
|
||||
|
||||
pipeline.setSource(std::make_shared<InvalidSource>(throw_every_nth_frame, num_frames));
|
||||
pipeline.start();
|
||||
while (has_frame) {
|
||||
++curr_frame;
|
||||
try {
|
||||
has_frame = pipeline.pull(cv::gout(out_mat));
|
||||
} catch (const std::exception& e) {
|
||||
EXPECT_TRUE(curr_frame % throw_every_nth_frame == 0);
|
||||
EXPECT_EQ(InvalidSource::exception_msg(), e.what());
|
||||
}
|
||||
}
|
||||
|
||||
// NB: Pull was called num_frames + 1(stop).
|
||||
EXPECT_EQ(num_frames, curr_frame - 1);
|
||||
}
|
||||
|
||||
} // namespace opencv_test
|
||||
|
||||
Reference in New Issue
Block a user