AlexeyAB / Yolo_mark

GUI for marking bounded boxes of objects in images for training neural network Yolo v3 and v2
https://github.com/AlexeyAB/darknet
The Unlicense
1.8k stars 680 forks source link

train vehicles recognition #181

Open ghost opened 4 years ago

ghost commented 4 years ago

Hi @AlexeyAB ,

Hope you are all well

I would like to retrain Yolov3 for only those class names:

person
car
motorcycle
bus
truck

Moreover, I want to remove pictures of trunk, car interiors, engine parts for cars, motorcycles, bus or truck.

My goal

Just recognize the shape of car, motorcycle, bus or truck from the outside

But it appears to be slow on my 69682 pictures, with yolo_mark, and I cannot browse quickly between images to remove the one correspondent to my custom filtering.

Is there a way to speed up the process ? Is there any alternative to it ?

Thanks in advance for an insights or inputs on these topics.

Cheers, X