Skip to content

Image postprocessor #720

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 19 commits into
base: master
Choose a base branch
from
Open

Image postprocessor #720

Changes from 1 commit
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
3b41352
init push
jonpsy Dec 3, 2021
0344650
Write main logic.
jonpsy Dec 3, 2021
99e709d
build file updated
jonpsy Dec 3, 2021
7c766c3
format file
jonpsy Dec 3, 2021
4cdbbf8
Use "is_input bool to switch input/output.
jonpsy Dec 9, 2021
2a39596
Preprocessor use new API
jonpsy Dec 9, 2021
ed455cb
single index instead of multi
jonpsy Dec 9, 2021
ae93793
Re-use logic from BuildImageTensorSpecs
jonpsy Dec 9, 2021
e24ef95
move in unknwn namespace
jonpsy Dec 11, 2021
f4b1607
move has_uint8_output_ to init()
jonpsy Dec 11, 2021
5236b8b
Use populate vector & rm size checks
jonpsy Dec 19, 2021
31c2b31
Pass tensor & metato tensor_spec. rm tensor count.
jonpsy Dec 22, 2021
213acb0
GetTensorMetada() checks subgraph of given meta
jonpsy Dec 22, 2021
9cdcb04
NormProcessUnit check in postprocessor to fallback
jonpsy Dec 23, 2021
352fcdb
store output tensor instead of its type.
jonpsy Dec 23, 2021
d1fa720
Remove is_input from image_tensor_specs.h
jonpsy Dec 23, 2021
63846f8
update BuildImageTensorSpec for preprocessor.
jonpsy Dec 23, 2021
8d9dc30
Fix compile error
jonpsy Dec 23, 2021
da766a4
Fix namespace
jonpsy Dec 23, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Fix namespace
  • Loading branch information
@jonpsy
jonpsy authored Dec 23, 2021
commit da766a495a14f46b2a0f628e30af63a4d3ead713
9 changes: 6 additions & 3 deletions tensorflow_lite_support/cc/task/vision/utils/image_tensor_specs.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,6 @@ StatusOr<const TensorMetadata*> GetTensorMetadataIfAny(
}
return tensor_metadata;
}
} // namespace

StatusOr<const ImageProperties*> GetImagePropertiesIfAny(
const TensorMetadata& tensor_metadata) {
Expand Down Expand Up @@ -126,12 +125,15 @@ StatusOr<absl::optional<NormalizationOptions>> GetNormalizationOptionsIfAny(
return normalization_options;
}

} // namespace

StatusOr<ImageTensorSpecs> BuildImageTensorSpecs(
const ModelMetadataExtractor& metadata_extractor,
const TensorMetadata* tensor_metadata, const TfLiteTensor* tensor) {
const ImageProperties* props = nullptr;
absl::optional<NormalizationOptions> normalization_options;
ASSIGN_OR_RETURN(auto metadata, GetTensorMetadataIfAny(metadata_extractor, tensor_metadata));
ASSIGN_OR_RETURN(const TensorMetadata* metadata,
GetTensorMetadataIfAny(metadata_extractor, tensor_metadata));
if (metadata != nullptr) {
ASSIGN_OR_RETURN(props, GetImagePropertiesIfAny(*metadata));
ASSIGN_OR_RETURN(normalization_options,
Expand Down Expand Up @@ -211,7 +213,8 @@ StatusOr<ImageTensorSpecs> BuildImageTensorSpecs(
if (bytes_size != height * width * depth * byte_depth) {
return CreateStatusWithPayload(
StatusCode::kInvalidArgument,
"The tensor in bytes does not correspond to the expected number of "
"The tensor size in bytes does not correspond to the expected number "
"of "
"pixels.",
TfLiteSupportStatus::kInvalidInputTensorSizeError);
}
Expand Down