Taking photo with custom camera iOS 11.0 Swift 4. Update error

In iOS 11, you should use like this :

@available(iOS 11.0, *)
func photoOutput(_ output: AVCapturePhotoOutput, didFinishProcessingPhoto photo: AVCapturePhoto, error: Error?) {
        let imageData = photo.fileDataRepresentation()
}