Viewing a single comment thread. View all comments

PaleontologistDue620 t1_j8s39ou wrote

do yourself a favor and go for the good old yolo, i think it has a tensorflow.js version too.

5

[deleted] OP t1_j8s73tb wrote

I am taking a look. But most tensorflow.js versions are either untrainable or too large, so I will probably use Keras and export the models as Layers, If I can manage to. I am still looking for light weight NNs, it's a challenge, so I may ask somewhere for candidates.

0

PaleontologistDue620 t1_j8s7qkn wrote

train it with either darknet or python then convert the weights if you need to, there are already scripts for everything you need to do, that's why i said YOLO in the first place.

3

[deleted] OP t1_j8s8e0x wrote

Sorry for the ignorance but wdym by darknet here? All I was planning is to use tensorflow's keras, as tf.js won't make it I think. I started a week ago so I am still digesting things.

1

PaleontologistDue620 t1_j8s9y0j wrote

you can train yolo on any framework you want and convert the weights later, then load them into your preferred inference framework. ( I'm not sure about js but in python you can load yolo models into opencv as well ). darknet is the original yolo framework which gives you scripts for training the model.

2

[deleted] OP t1_j8sdmmp wrote

Oh, that helps. I am not sure how true this is though. For example, TF.Keras and TF.SavedModel cant be converted into one another and have different features..Both can be used to predict but only one can be re trained and "tweaked" or extended from JS itself. And I am not sure you can convert Pytorch weights to Keras, but I will investigate. Apparently there is ONX that can be used to do it. I just dont want to train something that can not be converted and loaded into a browser.

What I learnt so far is that Sliding Window, Region of Interest, and Yolo are more like ways to prepare your data, and mostly any CNN could do the job, with more or less precision, I may be wrong. I am following this series https://www.youtube.com/watch?v=XXYG5ZWtjj0&list=PLhhyoLH6Ijfw0TpCTVTNk42NN08H6UvNq&index=2&ab_channel=AladdinPersson

0

LuckyNumber-Bot t1_j8sdnn3 wrote

All the numbers in your comment added up to 69. Congrats!

  5
+ 6
+ 42
+ 8
+ 6
+ 2
= 69

^(Click here to have me scan all your future comments.)
^(Summon me on specific comments with u/LuckyNumber-Bot.)

2

tedmobsky t1_j8v9fnl wrote

Good bot

1

B0tRank t1_j8v9glq wrote

Thank you, tedmobsky, for voting on LuckyNumber-Bot.

This bot wants to find the best and worst bots on Reddit. You can view results here.

^(Even if I don't reply to your comment, I'm still listening for votes. Check the webpage to see if your vote registered!)

1

[deleted] OP t1_j8wddyy wrote

It is interesting you commented this. I have spent a day yesterday trying to train some Yolo in python, but all the implementations on github are quite obsolete, apart from Yolov7.

Unless you refer to ultralytics only?

1

PaleontologistDue620 t1_j8wuk9b wrote

go for YOLO v3 or YOLO v4. i promise they'll be good enough for you, don't be bothered with version numbers . (if you need lighter models go for tiny versions of v3 and v4).

1

[deleted] OP t1_j8x413i wrote

I did it with tiny yolo v7, using google colab. My point is that there are barely any projects that are usable, unless you found some?

Yes the results were great, I am thinking of writing a little blogpost for others, it is actually quite simple because I found a tutorial in roboflow this time around.

Thanks for your support !

2

[deleted] OP t1_j90ysce wrote

Sorry to be annoying but I thought it was nice to give you some news as well. I was confused as to why there isnt yolo in pytorch, here it is why https://github.com/pytorch/vision/issues/6341

2

PaleontologistDue620 t1_j90z1yo wrote

no you're not annoying at all, thanks for the update :)

1

[deleted] OP t1_ja32trh wrote

Another update, I am reading the first yolo paper:

>We also train YOLO using VGG-16. This model is more accurate but also significantly slower than YOLO. It is useful for comparison to other detection systems that rely on VGG-16 but since it is slower than real-time the rest of the paper focuses on our faster models.

Which also explains that my main error was to use VGG16 without a good idea of how to make it understand where the objects are, which is what they did..

1

[deleted] OP t1_j8zb580 wrote

I have found out that VGG can indeed be used with SSD for the same task. Idk exactly what is the general idea but mostly you can combine CNNs with something else and get the bouding box. Pytorch has a SSD-VGG model.

I wonder why pytorch has no yolo implemented that we can just use..

1