Skip to content

Instantly share code, notes, and snippets.

@farukcankaya

farukcankaya/detectron2_visuzalization.py

Created October 3, 2022 19:03
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save farukcankaya/e788d9f338c47750477f5f45d5d0cfab to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save farukcankaya/e788d9f338c47750477f5f45d5d0cfab to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. farukcankaya created this gist Oct 3, 2022.
    20 changes: 20 additions & 0 deletions detectron2_visuzalization.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,20 @@
    def do_train_visualization(visualizer, model, inputs):
    training_mode = model.training
    model.eval()

    outputs = visualizer.inference(model, inputs)
    visualizer.process(inputs, outputs)

    model.train(training_mode)

    with EventStorage(start_iter) as storage:
    train_visualizer = VisualizationHook(cfg.DATASETS.TRAIN[0], storage=storage) # ADDED NEW

    for data, iteration in zip(data_loader, range(start_iter, max_iter)):
    storage.iter = iteration
    ...
    # ADDED NEW #############################################
    if iteration % 20 == 0: # log predictions every 20 iteration
    do_train_visualization(train_visualizer, model, inputs)
    ##########################################################
    ...