Image Classification
Transformers
Safetensors
English
Portuguese
puzzle_piece_sides_classifier
feature-extraction
puzzle
jigsaw-puzzle
multi-task-learning
dinov2
computer-vision
custom-model
custom_code
Instructions to use pablo-moreira/puzzle-piece-sides-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pablo-moreira/puzzle-piece-sides-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="pablo-moreira/puzzle-piece-sides-classifier", trust_remote_code=True) pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("pablo-moreira/puzzle-piece-sides-classifier", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
| { | |
| "auto_map": { | |
| "AutoImageProcessor": "puzzle_piece_sides_classifier_processor.PuzzlePieceSidesClassifierProcessor" | |
| }, | |
| "crop_size": { | |
| "height": 224, | |
| "width": 224 | |
| }, | |
| "do_center_crop": true, | |
| "do_convert_rgb": true, | |
| "do_normalize": true, | |
| "do_rescale": true, | |
| "do_resize": true, | |
| "image_mean": [ | |
| 0.48145466, | |
| 0.4578275, | |
| 0.40821073 | |
| ], | |
| "image_processor_type": "PuzzlePieceSidesClassifierProcessor", | |
| "image_std": [ | |
| 0.26862954, | |
| 0.26130258, | |
| 0.27577711 | |
| ], | |
| "pattern_id2label": { | |
| "0": "SSOI", | |
| "1": "SSIO", | |
| "10": "OSSO", | |
| "11": "SOOS", | |
| "12": "SOIO", | |
| "13": "SIOI", | |
| "14": "SIIO", | |
| "15": "SOOI", | |
| "16": "SOII", | |
| "17": "SIOO", | |
| "18": "SIII", | |
| "19": "SOOO", | |
| "2": "OSSI", | |
| "20": "OSOI", | |
| "21": "OSIO", | |
| "22": "OSII", | |
| "23": "ISIO", | |
| "24": "ISOO", | |
| "25": "ISOI", | |
| "26": "ISII", | |
| "27": "OSOO", | |
| "28": "OISI", | |
| "29": "OOSI", | |
| "3": "ISSO", | |
| "30": "OISO", | |
| "31": "IOSO", | |
| "32": "IOSI", | |
| "33": "IISI", | |
| "34": "IISO", | |
| "35": "OOSO", | |
| "36": "IIOS", | |
| "37": "IOIS", | |
| "38": "IOOS", | |
| "39": "OOIS", | |
| "4": "OISS", | |
| "40": "OIOS", | |
| "41": "OIIS", | |
| "42": "IIIS", | |
| "43": "OOOS", | |
| "44": "OIOI", | |
| "45": "IOII", | |
| "46": "OIII", | |
| "47": "IIOI", | |
| "48": "IIIO", | |
| "49": "IOIO", | |
| "5": "IOSS", | |
| "50": "IOOI", | |
| "51": "OIOO", | |
| "52": "OIIO", | |
| "53": "OOII", | |
| "54": "OOIO", | |
| "55": "IIOO", | |
| "56": "IIII", | |
| "57": "OOOO", | |
| "58": "IOOO", | |
| "59": "OOOI", | |
| "6": "SOIS", | |
| "7": "SIOS", | |
| "8": "SSOO", | |
| "9": "OOSS" | |
| }, | |
| "pattern_label2id": { | |
| "IIII": 56, | |
| "IIIO": 48, | |
| "IIIS": 42, | |
| "IIOI": 47, | |
| "IIOO": 55, | |
| "IIOS": 36, | |
| "IISI": 33, | |
| "IISO": 34, | |
| "IOII": 45, | |
| "IOIO": 49, | |
| "IOIS": 37, | |
| "IOOI": 50, | |
| "IOOO": 58, | |
| "IOOS": 38, | |
| "IOSI": 32, | |
| "IOSO": 31, | |
| "IOSS": 5, | |
| "ISII": 26, | |
| "ISIO": 23, | |
| "ISOI": 25, | |
| "ISOO": 24, | |
| "ISSO": 3, | |
| "OIII": 46, | |
| "OIIO": 52, | |
| "OIIS": 41, | |
| "OIOI": 44, | |
| "OIOO": 51, | |
| "OIOS": 40, | |
| "OISI": 28, | |
| "OISO": 30, | |
| "OISS": 4, | |
| "OOII": 53, | |
| "OOIO": 54, | |
| "OOIS": 39, | |
| "OOOI": 59, | |
| "OOOO": 57, | |
| "OOOS": 43, | |
| "OOSI": 29, | |
| "OOSO": 35, | |
| "OOSS": 9, | |
| "OSII": 22, | |
| "OSIO": 21, | |
| "OSOI": 20, | |
| "OSOO": 27, | |
| "OSSI": 2, | |
| "OSSO": 10, | |
| "SIII": 18, | |
| "SIIO": 14, | |
| "SIOI": 13, | |
| "SIOO": 17, | |
| "SIOS": 7, | |
| "SOII": 16, | |
| "SOIO": 12, | |
| "SOIS": 6, | |
| "SOOI": 15, | |
| "SOOO": 19, | |
| "SOOS": 11, | |
| "SSIO": 1, | |
| "SSOI": 0, | |
| "SSOO": 8 | |
| }, | |
| "pattern_labels": [ | |
| "SSOI", | |
| "SSIO", | |
| "OSSI", | |
| "ISSO", | |
| "OISS", | |
| "IOSS", | |
| "SOIS", | |
| "SIOS", | |
| "SSOO", | |
| "OOSS", | |
| "OSSO", | |
| "SOOS", | |
| "SOIO", | |
| "SIOI", | |
| "SIIO", | |
| "SOOI", | |
| "SOII", | |
| "SIOO", | |
| "SIII", | |
| "SOOO", | |
| "OSOI", | |
| "OSIO", | |
| "OSII", | |
| "ISIO", | |
| "ISOO", | |
| "ISOI", | |
| "ISII", | |
| "OSOO", | |
| "OISI", | |
| "OOSI", | |
| "OISO", | |
| "IOSO", | |
| "IOSI", | |
| "IISI", | |
| "IISO", | |
| "OOSO", | |
| "IIOS", | |
| "IOIS", | |
| "IOOS", | |
| "OOIS", | |
| "OIOS", | |
| "OIIS", | |
| "IIIS", | |
| "OOOS", | |
| "OIOI", | |
| "IOII", | |
| "OIII", | |
| "IIOI", | |
| "IIIO", | |
| "IOIO", | |
| "IOOI", | |
| "OIOO", | |
| "OIIO", | |
| "OOII", | |
| "OOIO", | |
| "IIOO", | |
| "IIII", | |
| "OOOO", | |
| "IOOO", | |
| "OOOI" | |
| ], | |
| "resample": 3, | |
| "rescale_factor": 0.00392156862745098, | |
| "side_id2label": { | |
| "0": "SMOOTH", | |
| "1": "OUTER", | |
| "2": "INNER" | |
| }, | |
| "side_label2id": { | |
| "INNER": 2, | |
| "OUTER": 1, | |
| "SMOOTH": 0 | |
| }, | |
| "side_labels": [ | |
| "SMOOTH", | |
| "OUTER", | |
| "INNER" | |
| ], | |
| "size": { | |
| "shortest_edge": 224 | |
| } | |
| } | |