Added swift image classifier test for quantized model
This commit is contained in:
parent
b354795d00
commit
d2780e4251
|
@ -28,6 +28,10 @@ class ImageClassifierTests: XCTestCase {
|
||||||
forResource: "mobilenet_v2_1.0_224",
|
forResource: "mobilenet_v2_1.0_224",
|
||||||
ofType: "tflite")
|
ofType: "tflite")
|
||||||
|
|
||||||
|
static let quantizedModelPath = bundle.path(
|
||||||
|
forResource: "mobilenet_v1_0.25_224_quant",
|
||||||
|
ofType: "tflite")
|
||||||
|
|
||||||
static let burgerImage = FileInfo(name: "burger", type: "jpg")
|
static let burgerImage = FileInfo(name: "burger", type: "jpg")
|
||||||
static let burgerRotatedImage = FileInfo(name: "burger_rotated", type: "jpg")
|
static let burgerRotatedImage = FileInfo(name: "burger_rotated", type: "jpg")
|
||||||
static let multiObjectsImage = FileInfo(name: "multi_objects", type: "jpg")
|
static let multiObjectsImage = FileInfo(name: "multi_objects", type: "jpg")
|
||||||
|
@ -258,6 +262,30 @@ class ImageClassifierTests: XCTestCase {
|
||||||
ImageClassifierTests.expectedResultsClassifyBurgerImageWithFloatModel)
|
ImageClassifierTests.expectedResultsClassifyBurgerImageWithFloatModel)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func testClassifyWithQuantizedModelSucceeds() throws {
|
||||||
|
|
||||||
|
let imageClassifierOptions =
|
||||||
|
try XCTUnwrap(
|
||||||
|
imageClassifierOptionsWithModelPath(
|
||||||
|
ImageClassifierTests.quantizedModelPath))
|
||||||
|
|
||||||
|
let imageClassifier = try XCTUnwrap(ImageClassifier(options: imageClassifierOptions))
|
||||||
|
|
||||||
|
let expectedCategories = [
|
||||||
|
ResultCategory(
|
||||||
|
index: 934,
|
||||||
|
score: 0.972656,
|
||||||
|
categoryName: "cheeseburger",
|
||||||
|
displayName: nil),
|
||||||
|
]
|
||||||
|
|
||||||
|
try assertResultsForClassifyImageWithFileInfo(
|
||||||
|
ImageClassifierTests.burgerImage,
|
||||||
|
usingImageClassifier: imageClassifier,
|
||||||
|
hasCategoryCount: ImageClassifierTests.mobileNetCategoriesCount,
|
||||||
|
andCategories: expectedCategories)
|
||||||
|
}
|
||||||
|
|
||||||
func testClassifyWithScoreThresholdSucceeds() throws {
|
func testClassifyWithScoreThresholdSucceeds() throws {
|
||||||
|
|
||||||
let imageClassifierOptions =
|
let imageClassifierOptions =
|
||||||
|
|
Loading…
Reference in New Issue
Block a user