()
| 247 | |
| 248 | @link_dist |
| 249 | def main(): |
| 250 | parser = argparse.ArgumentParser(description='Inference Solver') |
| 251 | parser.add_argument('--config', required=True, type=str, help="Prototype task yaml") |
| 252 | parser.add_argument('--recover', default="", help="Recover model path to visuazlie") |
| 253 | |
| 254 | parser.add_argument('-i', '--image_dir', required=True, dest="image_dir", type=str, |
| 255 | help="The image dir that you want to visuazlie.") |
| 256 | parser.add_argument('-m', '--meta_file', required=False, dest="meta_file", type=str, |
| 257 | help="The prototype custom meta file that you want to visuazlie. " |
| 258 | "If this argument are not provide, we will visualize the images in {image_dir}") |
| 259 | parser.add_argument('-o', '--output', default="./inference_resuts", dest="output", |
| 260 | help="the folder where results file or images will be saved.") |
| 261 | parser.add_argument('--visualize', default=True, help="Whether paint class and score on images to visualize.") |
| 262 | parser.add_argument('--sample', default=-1, type=float, |
| 263 | help="if gived number -1, remain all results. if 0 < gived number <=1, " |
| 264 | "sample {gived number * len(images_dir)} images, " |
| 265 | "if gived number > 1, sample gived number images.") |
| 266 | parser.add_argument('--cam', default=False, |
| 267 | help="Whether save gradcam results. See https://arxiv.org/abs/1610.02391 for details.") |
| 268 | parser.add_argument('--name', default="module.layer4", |
| 269 | help="the last feature extractor layer name you want to visualize gradcam results, " |
| 270 | "e.g. 'layer4' in resnet series.") |
| 271 | |
| 272 | args = parser.parse_args() |
| 273 | # build solver |
| 274 | inference_helper = Inference(args.__dict__) |
| 275 | inference_helper.inference() |
| 276 | |
| 277 | |
| 278 | if __name__ == '__main__': |
no test coverage detected