Views
No views yet
| Color (R,G,B) | Class Name (Persian) | Class Name (English) |
|---|---|---|
| (0, 0, 0) | باطله | Waste / Gangue |
| (100,150,100) | مگنتیت | Magnetite |
| (50,70,100) | پیریت | Pyrite |
| (100,60,60) | گوئتیت | Goethite |
| (60,30,80) | پیروتیت | Pyrrhotite |
| (100,30,10) | هماتیت | Hematite |
| (200,40,40) | کالکوپیریت | Chalcopyrite |
1from transformers import AutoImageProcessor, AutoModelForSemanticSegmentation
2from PIL import Image
3import torch
4
5processor = AutoImageProcessor.from_pretrained("fateme1380/segformer-mineral-500-final")
6model = AutoModelForSemanticSegmentation.from_pretrained("fateme1380/segformer-mineral-500-final")
7
8image = Image.open("your_mineral_image.jpg")
9inputs = processor(images=image, return_tensors="pt")
10with torch.no_grad():
11 outputs = model(**inputs)
12
13logits = outputs.logits
14pred = torch.nn.functional.interpolate(logits, size=image.size[::-1], mode="bilinear", align_corners=False)
15pred = pred.argmax(dim=1)[0].cpu().numpy()
16
17# حالا pred یه آرایه است که هر پیکسل مقدار 0 تا 6 داره (مطابق جدول بالا)