提交 2eba1984 编写于 作者: L liuruilong

remove mobilent ssd

上级 228d2e05
...@@ -21,7 +21,8 @@ import MetalPerformanceShaders ...@@ -21,7 +21,8 @@ import MetalPerformanceShaders
let platform: Platform = .GPU let platform: Platform = .GPU
let threadSupport = [1] let threadSupport = [1]
let modelHelperMap: [SupportModel : Runner] = [.mobilenet_ssd : Runner.init(inNet: MobileNet_ssd_hand.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform), //.mobilenet_ssd : Runner.init(inNet: MobileNet_ssd_hand.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform),
let modelHelperMap: [SupportModel : Runner] = [
.genet : Runner.init(inNet: Genet.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform), .genet : Runner.init(inNet: Genet.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform),
.mobilenet_ssd_ar : Runner.init(inNet: MobileNet_ssd_AR.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform)] .mobilenet_ssd_ar : Runner.init(inNet: MobileNet_ssd_AR.init(device: MetalHelper.shared.device), commandQueue: MetalHelper.shared.queue, inPlatform: platform)]
//, .genet : Genet.init() //, .genet : Genet.init()
...@@ -29,13 +30,14 @@ let modelHelperMap: [SupportModel : Runner] = [.mobilenet_ssd : Runner.init(inNe ...@@ -29,13 +30,14 @@ let modelHelperMap: [SupportModel : Runner] = [.mobilenet_ssd : Runner.init(inNe
enum SupportModel: String{ enum SupportModel: String{
// case mobilenet = "mobilenet" // case mobilenet = "mobilenet"
case mobilenet_ssd = "mobilenetssd" // case mobilenet_ssd = "mobilenetssd"
case genet = "genet" case genet = "genet"
case mobilenet_ssd_ar = "mobilenetssd_ar" case mobilenet_ssd_ar = "mobilenetssd_ar"
static func supportedModels() -> [SupportModel] { static func supportedModels() -> [SupportModel] {
//.mobilenet, // .mobilenet,
return [.genet, .mobilenet_ssd, .mobilenet_ssd_ar] // .mobilenet_ssd,
return [.genet, .mobilenet_ssd_ar]
} }
} }
......
...@@ -51,7 +51,7 @@ public class MobileNet_ssd_AR: Net{ ...@@ -51,7 +51,7 @@ public class MobileNet_ssd_AR: Net{
fatalError() fatalError()
} }
// let startDate = Date.init() let startDate = Date.init()
// print("scoreFormatArr: ") // print("scoreFormatArr: ")
//print((0..<score.capacity).map{ score.result[$0] }.strideArray()) //print((0..<score.capacity).map{ score.result[$0] }.strideArray())
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册