license: mit # Or apache-2.0, cc-by-sa-4.0, etc. - choose one | |
language: en | |
tags: | |
- pytorch | |
- computer-vision | |
- object-detection # Or more specific like instance-segmentation if you use masks | |
- 3d-dimensions | |
- pix3d | |
- resnet50 | |
# Add any other relevant tags | |
datasets: | |
- pix3d # The primary dataset used for training this specific model | |
# metrics: # If you have evaluation metrics like MSE or MAE | |
# - mse | |
# - mae | |
pipeline_tag: image-feature-extraction # Or a more custom tag if applicable | |
# widget: # If you were to create a widget for it (more advanced) | |
# - text: "Try with a sample image!" | |
# example_url: "URL_TO_A_SAMPLE_CROPPED_IMAGE_PATCH.jpg" | |