Internal change
PiperOrigin-RevId: 523916108
This commit is contained in:
parent
d96b6e7ed9
commit
c8b9cec7ef
|
@ -20,7 +20,7 @@
|
||||||
namespace mediapipe {
|
namespace mediapipe {
|
||||||
|
|
||||||
// Uniform normalization to 0-255.
|
// Uniform normalization to 0-255.
|
||||||
uint8 FlowQuantizerModel::Apply(const float val, const int channel) const {
|
uint8_t FlowQuantizerModel::Apply(const float val, const int channel) const {
|
||||||
CHECK_LT(channel, model_.min_value_size());
|
CHECK_LT(channel, model_.min_value_size());
|
||||||
const auto& min_value = model_.min_value(channel);
|
const auto& min_value = model_.min_value(channel);
|
||||||
const auto& max_value = model_.max_value(channel);
|
const auto& max_value = model_.max_value(channel);
|
||||||
|
@ -31,7 +31,7 @@ uint8 FlowQuantizerModel::Apply(const float val, const int channel) const {
|
||||||
} else if (res > 1.0) {
|
} else if (res > 1.0) {
|
||||||
res = 1.0;
|
res = 1.0;
|
||||||
}
|
}
|
||||||
return static_cast<uint8>(res * 255);
|
return static_cast<uint8_t>(res * 255);
|
||||||
}
|
}
|
||||||
|
|
||||||
void FlowQuantizerModel::LoadFromProto(const QuantizerModelData& data) {
|
void FlowQuantizerModel::LoadFromProto(const QuantizerModelData& data) {
|
||||||
|
|
Loading…
Reference in New Issue
Block a user