README.md

metadata

tags:
  - computer_vision
  - vision_models_playground
  - custom-implementation

Vision Models Playground

This is a trained model from the Vision Models Playground repository. Link to the repository: https://github.com/Akrielz/vision_models_playground

Model

This model is a custom implementation of ResNetYoloV1 from the vision_models_playground.models.segmentation.yolo_v1 module. Please look in the config file for more information about the model architecture.

Metrics

The model was evaluated on the following dataset: YoloPascalVocDataset from vision_models_playground.datasets.yolo_pascal_voc_dataset

These are the results of the evaluation:

• MulticlassAccuracy: 0.7241
• MulticlassAveragePrecision: 0.7643
• MulticlassAUROC: 0.9684
• Dice: 0.7241
• MulticlassF1Score: 0.7241
• LossTracker: 4.1958

Additional Information

The train and evaluation runs are also saved using tensorboard. You can use the following command to visualize the runs:

tensorboard --logdir ./model

tensorboard --logdir ./eval