search

sacmehta / ESPNetv2

A light-weight, power efficient, and general purpose convolutional neural network
MIT License
447 stars 70 forks source link

Question about predict on espnetv2 #31

Closed marcusvlc closed 5 years ago

marcusvlc commented 5 years ago

How can I be using ESPNETv2 to make inferences on random images? In your repository and in the Edgenets repository I just found ways to evaluate in a specific dataset, but I would like to pass external images as a parameter and see the net result, how can I be doing this?

sacmehta commented 5 years ago

How hard is to adapt the test_segmentation.py file in the EdgeNets repository?

marcusvlc commented 5 years ago

A little difficult since I never use pytorch. I realized that you have the .onnx templates, would you have them in tensorflow .pb? I am trying to do this conversion with onnx but am having some errors:

ValueError: Dimensions must be equal, but are 32 and 112 for 'mul' (op: 'Mul') with input shapes: [32], [1,32,112,112].

sacmehta commented 5 years ago

I am not sure if we will have tensorflow .pb files in the near future. However, the mentioned error is because of incorrect mapping of the PRelu layer. Please ask on the appropriate forums