public abstract class TrainedModelDetector extends Object implements Detector
Constructor and Description |
---|
TrainedModelDetector() |
Modifier and Type | Method and Description |
---|---|
MediaType |
detect(InputStream input,
Metadata metadata)
Detects the content type of the given input document.
|
int |
getMinLength() |
abstract void |
loadDefaultModels(ClassLoader classLoader) |
void |
loadDefaultModels(File modelFile) |
abstract void |
loadDefaultModels(InputStream modelStream) |
void |
loadDefaultModels(Path modelFile) |
protected float[] |
readByteFrequencies(InputStream input)
Read the
inputstream and build a byte frequency histogram |
protected void |
registerModels(MediaType type,
TrainedModel model) |
public int getMinLength()
public MediaType detect(InputStream input, Metadata metadata) throws IOException
Detector
application/octet-stream
if the type of the document
can not be detected.
If the document input stream is not available, then the first
argument may be null
. Otherwise the detector may
read bytes from the start of the stream to help in type detection.
The given stream is guaranteed to support the
mark feature
and the detector
is expected to mark
the stream before
reading any bytes from it, and to reset
the stream before returning. The stream must not be closed by the
detector.
The given input metadata is only read, not modified, by the detector.
detect
in interface Detector
input
- document input stream, or null
metadata
- input metadata for the documentapplication/octet-stream
IOException
- if the document input stream could not be readprotected float[] readByteFrequencies(InputStream input) throws IOException
inputstream
and build a byte frequency histograminput
- stream to read fromIOException
public void loadDefaultModels(Path modelFile)
public void loadDefaultModels(File modelFile)
public abstract void loadDefaultModels(InputStream modelStream)
public abstract void loadDefaultModels(ClassLoader classLoader)
protected void registerModels(MediaType type, TrainedModel model)
Copyright © 2007–2021 The Apache Software Foundation. All rights reserved.