about testing

[Leo551]

Thanks for your work. I have a question that after training how to test the model on my own data sets?

[HasnainRaz]

You can set the mode flag to infer and provide the test images directory path like so:

python main.py --mode=infer --infer_data=path/to/infer/images --output_folder=path/to/where/you/want/the/predicted/images/to/be/saved

You might have to modify the infer function in model.py, since currently it is a barebones predict function.