Darknet: train and test set in YOLO v3

Created on 20 Apr 2020  路  4Comments  路  Source: pjreddie/darknet

my dataset contains few images (~490 images for 2 classes). so it is fine if don't split the dataset into train and test set and use all the images for training? Does the size of test set affect the training of yolo?

Most helpful comment

yes there is a functionality
./darknet detector map data/YourDataFile.data cfg/yourCfgFile.cfg backup/YourWeightsFile.weights -points 0 -thresh 0.1
in data file you need to mention the path of your test set then it will go through all the images and annotations and compare them to the detected classes make sure these images are not from training set give completely new images to find the real accuracy.

All 4 comments

Nope, it does not affect training but while calculating the accuracy for your created model then you might need to provide the images which are never shown to the model to check how accurately it is getting predictions while calculating mAP other than that there is a no use of test and train set division

and how do we calculate accuracy in yolo? I just know how to detect the object, but is there some functionality to test the accuracy given a validation dataset?

yes there is a functionality
./darknet detector map data/YourDataFile.data cfg/yourCfgFile.cfg backup/YourWeightsFile.weights -points 0 -thresh 0.1
in data file you need to mention the path of your test set then it will go through all the images and annotations and compare them to the detected classes make sure these images are not from training set give completely new images to find the real accuracy.

Hello @Rajasekhar06 , I am training with 5 species and gave the command you mentioned above and I am getting the results shown in the below image , what exactly is the meaning of TP ,FP and FN for each class ?

all

Was this page helpful?
0 / 5 - 0 ratings

Related issues

sujithm picture sujithm  路  3Comments

ryuzakinho picture ryuzakinho  路  4Comments

TheHidden1 picture TheHidden1  路  3Comments

gpsmit picture gpsmit  路  3Comments

cadip92 picture cadip92  路  3Comments