PyTorch Geometric (PyG) 主要設計用于處理圖結構數據,而不是圖像處理。它提供了用于圖數據處理的工具,如圖卷積網絡、數據加載和預處理等。以下是PyG的主要用途和特點:
雖然PyTorch Geometric (PyG) 不是為圖像處理設計的,但PyTorch本身提供了強大的圖像處理庫,如torchvision
,它包含了一系列用于圖像預處理、增強和加載的工具。如果你需要進行圖像處理,建議使用torchvision
庫。
總的來說,PyTorch Geometric (PyG) 不適合用于圖像處理,而應該使用PyTorch的torchvision
庫來處理圖像數據。