【发布时间】:2019-12-23 00:07:00
【问题描述】:
我是 Flutter 和 Dart 的新手,我正在尝试使用 XIAOMI MI 9 SE (ANDROID) 上的 Camera Plugin 来检测我的 TensorFlow Lite 模型上的对象。一切都在纵向模式下工作,但是当我将智能手机打开横向视图时,预览不会旋转,保持纵向的方向并且看起来不正确。
我尝试使用其他版本的相机插件,但没有用。
detectObjectOnFrame 代码:
` Tflite.detectObjectOnFrame(
bytesList: img.planes.map((plane) {
return plane.bytes;
}).toList(),
model: widget.model == yolo ? "YOLO" : "SSDMobileNet",
imageHeight: img.height,
imageWidth: img.width,
imageMean: widget.model == yolo ? 0 : 127.5,
imageStd: widget.model == yolo ? 255.0 : 127.5,
numResultsPerClass: 1,
threshold: widget.model == yolo ? 0.2 : 0.4,
).then((recognitions) {
int endTime = new DateTime.now().millisecondsSinceEpoch;
print("Detection took ${endTime - startTime}");
widget.setRecognitions(recognitions, img.height, img.width);
isDetecting = false;`
CAMERA.DART 文件:
final List cameras;
final Callback setRecognitions;
final String model;
Camera(this.cameras, this.model, this.setRecognitions);
_CameraState createState() => new _CameraState();
}
class _CameraState extends State {
CameraController controller;
bool isDetecting = false;
void initState() {
super.initState();
if (widget.cameras == null || widget.cameras.length < 1) {
print('No camera is found');
} else {
controller = new CameraController(
widget.cameras[0],
ResolutionPreset.high,
);
controller.initialize().then((_) {
if (!mounted) {
return;
}
setState(() {});
controller.startImageStream((CameraImage img) {
if (!isDetecting) {
isDetecting = true;
int startTime = new DateTime.now().millisecondsSinceEpoch;
if (widget.model == mobilenet) {
..........
.......... (Ignored code)
isDetecting = false;
});
} else {
Tflite.detectObjectOnFrame(
..........
.......... (Ignored code)
});
}
}
});
});
}
}
void dispose() {
controller?.dispose();
super.dispose();
}
Widget build(BuildContext context) {
if (controller == null || !controller.value.isInitialized) {
return Container();
}
var tmp = MediaQuery.of(context).size;
var screenH = math.max(tmp.height, tmp.width);
var screenW = math.min(tmp.height, tmp.width);
tmp = controller.value.previewSize;
var previewH = math.max(tmp.height, tmp.width);
var previewW = math.min(tmp.height, tmp.width);
var screenRatio = screenH / screenW;
var previewRatio = previewH / previewW;
return OverflowBox(
maxHeight:
screenRatio > previewRatio ? screenH : screenW / previewW * previewH,
maxWidth:
screenRatio > previewRatio ? screenH / previewH * previewW : screenW,
child: CameraPreview(controller),
);
}
}
HOME.DART:
setState(() {
_model = model;
});
loadModel();
}
setRecognitions(recognitions, imageHeight, imageWidth) {
setState(() {
_recognitions = recognitions;
_imageHeight = imageHeight;
_imageWidth = imageWidth;
});
}
@override
Widget build(BuildContext context) {
Size screen = MediaQuery.of(context).size;
return Scaffold(
body: _model == ""
? Center(
child: Column(
mainAxisAlignment: MainAxisAlignment.center,
children: [
RaisedButton(
child: const Text(yolo),
onPressed: () => onSelect(yolo),
),
],
),
)
: Stack(
children: [
Camera(
widget.cameras,
_model,
setRecognitions,
),
BndBox(
_recognitions == null ? [] : _recognitions,
math.max(_imageHeight, _imageWidth),
math.min(_imageHeight, _imageWidth),
screen.height,
screen.width,
_model),
],
),
);
}
}
Pubspec.yaml:
flutter_test:
sdk: flutter
camera: ^0.5.3
tflite: 1.0.4
此代码在横向视图中显示不正确的图像,预览无法与智能手机一起转动。
感谢任何帮助。谢谢。
【问题讨论】:
标签: android flutter android-camera screen-orientation android-camera2