tensorflow-java-examples-spring icon indicating copy to clipboard operation
tensorflow-java-examples-spring copied to clipboard

Can we adapt the server to Yolo v3? How to get proper .pb file?

Open locoda opened this issue 5 years ago • 4 comments

Since I noticed that your project used .pb file for graph, how you get that? We tried several projects, and got errors like Op type not registered 'LeakyRelu' in binary running on xxx. Make sure the Op and Kernel are registered in the binary running in this process. How you get a proper .pb file?

locoda avatar Apr 19 '19 16:04 locoda

Hi, Unfortunately, the current version does not support the YoloV3 model. The current version uses the following classifier to parse the output tensor and to display it on the output stream: https://github.com/szaza/tensorflow-java-examples-spring/blob/master/src/main/java/edu/ml/tensorflow/service/classifier/YOLOClassifier.java; A similar classifier should be written for the YoloV3. The current classifier has been implemented based on the Yolo9000 article: https://arxiv.org/pdf/1612.08242.pdf; To implement the YoloV3 classifier please have a look at the following article: https://pjreddie.com/media/files/papers/YOLOv3.pdf;

szaza avatar Apr 20 '19 07:04 szaza

Unfortunately, in the current days I do not have too much time to add yolov3 support for this project, however I would be very happy if I could review a pull request with this new feature. I would also provide you some instructions to develop this feature. For this reason I created a new branch: https://github.com/szaza/tensorflow-java-examples-spring/tree/feature/add-yolov3-support; I would be happy if I could add you as a contributor to the project. @spaul13 already started to implement it and I believe we could develop it together. If anybody else is interested in this feature, please let me know and I'll add them as a contributor.

szaza avatar Apr 20 '19 09:04 szaza

Could you support GPU computing under windows10 system?

swzaaaaaaa avatar Apr 30 '20 15:04 swzaaaaaaa

Hi, I have question personally... I have my yolo model and I know python but not JAVA. I want to connect my TF by JAVA laguage... What should I study? your code is good reference for me, I should change "https://github.com/szaza/tensorflow-java-examples-spring/blob/master/src/main/java/edu/ml/tensorflow/service/classifier/YOLOClassifier.java" to custom as my yolo model? (TF input, output connecting by JAVA) (Also, actually my code is pytorch...;;)

NeighborhoodCoding avatar Sep 10 '20 04:09 NeighborhoodCoding