uchidama / MNIST-TrainDataForYOLO

Make MNIST train data for YOLO.
MIT License
20 stars 9 forks source link

program works but i don't know where is the prediction result #3

Open thisisnottaken opened 2 years ago

thisisnottaken commented 2 years ago

dear sir, I am looking for translating image of single digit to text . Your code is very nice . My main purpose is to use pretrained weights because my laptop has 4GB RAM and no GPU . here is the output on feeding an input-image ~/darknet$ ./darknet detector test cfg/voc-mnist.data cfg/tiny-yolo-mnist.cfg weights/tiny-yolo-mnist_500000.weights my-drawing.png layer filters size input output 0 conv 16 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 16 0.150 BFLOPs 1 max 2 x 2 / 2 416 x 416 x 16 -> 208 x 208 x 16 2 conv 32 3 x 3 / 1 208 x 208 x 16 -> 208 x 208 x 32 0.399 BFLOPs 3 max 2 x 2 / 2 208 x 208 x 32 -> 104 x 104 x 32 4 conv 64 3 x 3 / 1 104 x 104 x 32 -> 104 x 104 x 64 0.399 BFLOPs 5 max 2 x 2 / 2 104 x 104 x 64 -> 52 x 52 x 64 6 conv 128 3 x 3 / 1 52 x 52 x 64 -> 52 x 52 x 128 0.399 BFLOPs 7 max 2 x 2 / 2 52 x 52 x 128 -> 26 x 26 x 128 8 conv 256 3 x 3 / 1 26 x 26 x 128 -> 26 x 26 x 256 0.399 BFLOPs 9 max 2 x 2 / 2 26 x 26 x 256 -> 13 x 13 x 256 10 conv 512 3 x 3 / 1 13 x 13 x 256 -> 13 x 13 x 512 0.399 BFLOPs 11 max 2 x 2 / 1 13 x 13 x 512 -> 13 x 13 x 512 12 conv 1024 3 x 3 / 1 13 x 13 x 512 -> 13 x 13 x1024 1.595 BFLOPs 13 conv 512 3 x 3 / 1 13 x 13 x1024 -> 13 x 13 x 512 1.595 BFLOPs 14 conv 75 1 x 1 / 1 13 x 13 x 512 -> 13 x 13 x 75 0.013 BFLOPs 15 detection mask_scale: Using default '1.000000' Loading weights from weights/tiny-yolo-mnist_500000.weights...Done! my-drawing.png: Predicted in 1.729245 seconds.

can you tell me where is the predicted value ?

Thanks. Let me know if any more information is required .