Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / api / CPP / prior_box.hpp
index a21afda..c5ad40a 100644 (file)
@@ -91,6 +91,9 @@ struct prior_box : public primitive_base<prior_box, CLDNN_PRIMITIVE_DESC(prior_b
             if (!already_exist) {
                 this->aspect_ratios.push_back(new_aspect_ratio);
                 if (flip) {
+                    if (std::fabs(new_aspect_ratio) < std::numeric_limits<float>::epsilon()) {
+                        throw std::runtime_error("prior_box aspect ratio can't be zero!");
+                    }
                     this->aspect_ratios.push_back(1.f / new_aspect_ratio);
                 }
             }