search

hollance / Forge

A neural network toolkit for Metal
MIT License
1.27k stars 172 forks source link

Custom TinyYOLO doesn't work #37

Closed MatinC closed 6 years ago

MatinC commented 6 years ago

I have been trying to use Forge to detect a custom object based on TinyYolo model. By the way, every time I tried to run it gave me this error. My custom model consists of 2 classes, and tried to changes several parameters and seems not to work

untitled

The TinyYolo which based on voc that came with your example works fine.

hollance commented 6 years ago

It looks like your output array has more or fewer than 71313 numbers in it.

MatinC commented 6 years ago

I've change it to 35 and it worked Thanks you