AI說書 - 從0開始 - 571 | Swin2ForImageClassification 推論

我想要一天分享一點「LLM從底層堆疊的技術」，並且每篇文章長度控制在三分鐘以內，讓大家不會壓力太大，但是又能夠每天成長一點。

!curl -L https://raw.githubusercontent.com/Denis2054/Transformers_3rd_Edition/master/Chapter18/car_in_night.jpg --output "car_in_night.jpg"
from PIL import Image
image_path = "/content/car_in_night.jpg"
image = Image.open(image_path)
image

結果為：

再執行一次推論，程式為：

model_name = "autotrain-training-cifar-10-81128141657"
output = query("car_in_night.jpg", model_name)
classify_image(output)

結果為：