系统环境:Win11+Python3.9.7
1.介绍
torchvision 是PyTorch中专门用来处理图像的库
2.问题
运行程序时出现No module named ‘torchvision‘错误
3.解决方案
(1)pip安装torchvision
pip install torchvision
(2)pip安装指定版本
pip install torchvision==0.1.8
4.备注
Package Reference
Transforming and augmenting images
Scriptable transforms
Compositions of transforms
Transforms on PIL Image and torch.*Tensor
Transforms on PIL Image only
Transforms on torch.*Tensor only
Conversion Transforms
Generic Transforms
Automatic Augmentation Transforms
Functional Transforms
Models and pre-trained weights
Classification
Semantic Segmentation
Object Detection, Instance Segmentation and Person Keypoint Detection
Video classification
Optical flow
Datasets
Built-in datasets
Base classes for custom datasets
Utils
draw_bounding_boxes
draw_segmentation_masks
draw_keypoints
flow_to_image
make_grid
save_image
Operators
batched_nms
box_area
box_convert
box_iou
clip_boxes_to_image
deform_conv2d
generalized_box_iou
generalized_box_iou_loss
masks_to_boxes
nms
ps_roi_align
ps_roi_pool
remove_small_boxes
roi_align
roi_pool
sigmoid_focal_loss
stochastic_depth
RoIAlign
PSRoIAlign
RoIPool
PSRoIPool
DeformConv2d
MultiScaleRoIAlign
FeaturePyramidNetwork
StochasticDepth
FrozenBatchNorm2d
SqueezeExcitation
Reading/Writing images and videos
Video
Fine-grained video API
Image
Feature extraction for model inspection
API Reference