Views
No views yet
1pip install -U vesuvius transformer timesformer-pytorch
2vesuvius.accept_terms --yes1import vesuvius
2import numpy as np
3import matplotlib.pyplot as plt
4from vesuvius import Volume
5from transformers import pipeline
6
7
8segment = Volume(20230827161847, normalize=True)
9
10classifier = pipeline(model="scrollprize/ink_detection_pipeline", trust_remote_code=True, tile_size=64, stride=32, scale_factor=16)
11prediction = classifier(np.transpose(segment[17:43, 200:5600,1000:4600],(1,2,0)).astype(np.float32)) # prediction shape ~ (512,512)
12
13plt.imshow(prediction)