cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 525358261
This commit is contained in:
MediaPipe Team 2023-04-18 23:29:39 -07:00 committed by Copybara-Service
parent d7039c90dc
commit 8c8ba9511a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mediapipe/modules/objectron/calculators/frame_annotation_to_timed_box_list_calculator.cc
View File

@ -91,8 +91,8 @@ absl::Status FrameAnnotationToTimedBoxListCalculator::Process(
TimedBoxProto* added_box = output_objects->add_box(); TimedBoxProto* added_box = output_objects->add_box();
ComputeBoundingRect(key_points, added_box); ComputeBoundingRect(key_points, added_box);
added_box->set_id(annotation.object_id()); added_box->set_id(annotation.object_id());
const int64 time_msec = const int64_t time_msec =
static_cast<int64>(std::round(frame_annotation.timestamp() / 1000)); static_cast<int64_t>(std::round(frame_annotation.timestamp() / 1000));
added_box->set_time_msec(time_msec); added_box->set_time_msec(time_msec);
} }

4
mediapipe/modules/objectron/calculators/tflite_tensors_to_objects_calculator.cc
View File

@ -76,7 +76,7 @@ class TfLiteTensorsToObjectsCalculator : public CalculatorBase {
// In a single MediaPipe session, the IDs are unique. // In a single MediaPipe session, the IDs are unique.
// Also assign timestamp for the FrameAnnotation to be the input packet // Also assign timestamp for the FrameAnnotation to be the input packet
// timestamp. // timestamp.
void AssignObjectIdAndTimestamp(int64 timestamp_us, void AssignObjectIdAndTimestamp(int64_t timestamp_us,
FrameAnnotation* annotation); FrameAnnotation* annotation);
int num_classes_ = 0; int num_classes_ = 0;
@ -207,7 +207,7 @@ void TfLiteTensorsToObjectsCalculator::Project3DTo2D(
} }
void TfLiteTensorsToObjectsCalculator::AssignObjectIdAndTimestamp( void TfLiteTensorsToObjectsCalculator::AssignObjectIdAndTimestamp(
int64 timestamp_us, FrameAnnotation* annotation) { int64_t timestamp_us, FrameAnnotation* annotation) {
for (auto& ann : *annotation->mutable_annotations()) { for (auto& ann : *annotation->mutable_annotations()) {
ann.set_object_id(GetNextObjectId()); ann.set_object_id(GetNextObjectId());
} }