PGN(Patch Generation Network)
Train
python examples/train.py --config-file /path/to/config_file.yml
If you want to output DEBUG
level logs, you can modify INFO.LOG_LEVEL
option in the config file or use:
python examples/train.py --config-file /path/to/config_file.yml INFO.LOG_LEVEL DEBUG
Logger with DEBUG
level will output most of time consuming information.
Eval
Simply get patches for a image:
python examples/eval_single.py --config-file /path/to/config_file.yml --image_path /path/to/image.jpg
If you want to show recall evaluation result, just add --load_panda_annotation
flag.
python examples/eval_single.py --config-file /path/to/config_file.yml --image_path /path/to/image.jpg --load_panda_annotation
The annotation info for PANDA test set will be loaded according to the config in config_file
, so be sure the image_path
refers to an image in PANDA test set.
Visualize
If you want to visualize the patches in the image, just add --display
flag.
python examples/eval_single.py --config-file /path/to/config_file.yml --image_path /path/to/image.jpg --display
If the operation system does not support the display mode, you can user the --save_path
parameter to save result into a file:
python examples/eval_single.py --config-file /path/to/config_file.yml --image_path /path/to/image.jpg --save_path result.jpg