Premiär! Fyndchans i SweClockers Månadens Drop

Deeplearning4J + YOLO - Bildigenkänning med detektion

Permalänk

Deeplearning4J + YOLO - Bildigenkänning med detektion

Hej!

Jag vill börja med bildigenkänning + detektion, dvs det där som Tesla har i sina bilar för att avgöra om det är t.ex. en katt eller en tant man kör på.

Då finns det ett populärt bibliotek som kallas för Deeplearning4J. Tensorflow och Keras finns också. Men jag ska bygga en applikation så då är Deeplearning4J något mer lämpat än vetenskapliga studier med Python.

Jag har en fråga. Jag tänker använda mig av något som heter YOLO2. Alltså det är en hyfsat modern version av YOLO - You only look once.
Senaste är YOLO4 och den som många använder är YOLO3. Men YOLO4 börjar snart bli beprövad.

Men Deeplearning4J har YOLO2 hur som helst.
Är det någon här som har ett förslag på hur jag använder en befintlig tränad YOLO2 modell med Deeplearning4J?
Jag har inte lust att träna egen YOLO2 modell då det krävs liksom 1000 tals bilder per objekt, eller MER!

Visst, man kan använda 100 bilder per objekt om man vill, man får väldigt dålig precision då. Dessutom de som tränar YOLO modeller är riktigt duktiga dataforskare också. Inte några nybörjare som har tittat 10 minuter på Youtube.

Permalänk
Medlem

Ett tag sedan denna tråden skapades så du kanske har hittat vad du söker nu men:

Det finns ju flera alternativ när det gäller weights+cfg på deras hemsida att använda, har du testat dom? https://pjreddie.com/darknet/yolov2/

Eller kör: git clone https://github.com/pjreddie/darknet + wget https://pjreddie.com/media/files/yolov3.weights

Bara att ladda ner & köra, sen kryllar det ju av dataset från både Facebook, Google & flera andra stora företag, kryllar av dataset & modeller på Github

Sen för D4J så finns det ju: https://deeplearning4j.org/api/latest/org/deeplearning4j/zoo/...

Och D4J dokumentation är ju ganska välstrukturerad & lättförstådd så jag vet inte riktigt vad det är du frågar efter egentligen för allt är ju redan serverat på ett silverfat till dig?

Iaf, skall du köra YOLO så kör det på GPU, senast jag körde det så fick jag ut ca 0.3FPS på en 4 kärnig VM & +30-60fps på GPU (2080), skall du köra på CPU så kör något genom OpenCV4.X eller liknande, YOLO är tungt

Visa signatur

Hur kan syltkakor överleva i det vilda utan ögon?

Permalänk
Medlem

@heretic16: Jag skulle råda dig att köra på YoloV3 eller V4, speciellt YoloV4 om du vill ha en bra detektor.
Tänkte du för ett eller några enskilda typer av objekt, eller vill du detektera en massa olika?

Jag har själv nyligen börjat med detta men har tränat en modell att detektera bilar ovanifrån i YoloV4, tränad via Darknet.
Skriv gärna här eller på PM om du vill ha hjälp.

Var inte rädd för att skapa en egen modell (weight), det är mycket mycket bättre än pre-tränade, men det kräver tid, dock inte så mycket kunskap.