thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Training complete, but can't test the neural net #1127 #1128

Closed ankitAMD closed 4 years ago

ankitAMD commented 4 years ago

Checkpoint is new file.... but ....you have to use .meta file you have to use .pb file for testing.

When you train the model on terminal you have to add in your command "--savepb " which can save your .pb file in your built_graph folder.

    Command Example 

            "python flow  --model cfg/yolo-1c.cfg --load bin/yolo.weights --train --annotation new_model_data/annotations --dataset new_model_data/images  --epoch 40 --savepb" 

After training two new files created in the form of .meta and .pb extension in the built_graph folder. Use these two files(.meta and .pb extension files) after training in your testing code of Yolo project.

Use this link to see how to use ............

https://github.com/ankitAMD/Darkflow-object-detection/blob/master/Custom%20Automated%20Testing%20images%20of%20Solar_Panel%20(using%20tiny-yolo.cfg%20and%20weights).ipynb

Your new weight file are .meta and new configuration file are in .pb form extension.

I think your issues are cleared.Please like My comment and close both issues after solved.

Originally posted by @ankitAMD in https://github.com/thtrieu/darkflow/issues/1127#issuecomment-575050229