Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.
6577 토론

NCS1: Fine-tune TinyYOLO model for custom dataset

idata
직원
1,221 조회수

I was wondering if there was an easy way to fine-tune the TinyYOLO model to detect objects from a custom dataset in real time with bounding boxes and then compile the Caffe files into the NCS graph file. There are 4 classes in my dataset with about 150 images for each class so it is not a big dataset.

0 포인트
3 응답
idata
직원
1,044 조회수

@user123

 

I trained with a custom dataset based on the following repository.

 

https://github.com/duangenquan/YoloV2NCS.git
0 포인트
idata
직원
1,044 조회수

What file in the repository do I need to edit to have it train on my own dataset?

0 포인트
idata
직원
1,044 조회수

And I should probably place bounding boxes on the objects in my images? If so, what did you use to do that? Thanks

0 포인트
응답