search

mortenjust / tensorswift-ios

Using Tensorflow from Swift
43 stars 4 forks source link

Object Detection #3

Open eweill opened 7 years ago

eweill commented 7 years ago

Are you aware of an easy way to modify the image recognition portions of this project for object detection instead?

I have many networks that work with object detection and would like to port them to iOS.

Thanks.

mortenjust commented 7 years ago

Sounds interesting. Have you tried swapping out the model with your own and changing the code to match your output layer?