Skip to content

Commit

Permalink
clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
freedomtan committed Feb 1, 2024
1 parent a7ff320 commit 49d73aa
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 14 deletions.
10 changes: 5 additions & 5 deletions flutter/assets/tasks.pbtxt
Original file line number Diff line number Diff line change
Expand Up @@ -247,8 +247,8 @@ task {
id: "MobileNetEdgeTPUv2-L"
name: "MobileNetEdgeTPUv2-L"
offset: 0
image_width: 224
image_height: 224
image_width: 384
image_height: 384
num_classes: 1000
}
}
Expand Down Expand Up @@ -283,9 +283,9 @@ task {
model {
id: "MobileNetEdgeTPUv2-L"
name: "MobileNetEdgeTPUv2-L"
offset: 1
image_width: 224
image_height: 224
offset: 0
image_width: 384
image_height: 384
num_classes: 1000
}
}
8 changes: 0 additions & 8 deletions flutter/cpp/datasets/imagenet.cc
Original file line number Diff line number Diff line change
Expand Up @@ -62,18 +62,10 @@ Imagenet::Imagenet(Backend *backend, const std::string &image_dir,
// Prepares the preprocessing stage.
tflite::evaluation::ImagePreprocessingConfigBuilder builder(
"image_preprocessing", DataType2TfType(input_format_.at(0).type));
#if 0
builder.AddResizingStep(image_width / kCroppingFraction,
image_height / kCroppingFraction, true);
builder.AddCroppingStep(image_width, image_height, false);
#endif
#if 1
builder.AddCroppingStep(kCroppingFraction, true);
builder.AddResizingStep(image_width, image_height, false);
#endif

builder.AddDefaultNormalizationStep();

preprocessing_stage_.reset(
new tflite::evaluation::ImagePreprocessingStage(builder.build()));
if (preprocessing_stage_->Init() != kTfLiteOk) {
Expand Down
2 changes: 1 addition & 1 deletion flutter/integration_test/expected_throughput.dart
Original file line number Diff line number Diff line change
Expand Up @@ -178,7 +178,7 @@ const Map<String, Map<String, Interval>> _imageClassificationOffline = {
_kPixelBackend: {
_kPixel6: Interval(min: 1000, max: 1700),
},
_kQtiBackend: {
_kQtiBackend: {
_kS22Ultra: Interval(min: 2600, max: 3000),
},
_kMediatekBackend: {
Expand Down

0 comments on commit 49d73aa

Please sign in to comment.