Mask R-CNN instance segmentation training code
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
# Mask R-CNN 实例分割训练依赖
|
||||
# Python 3.11
|
||||
|
||||
# PyTorch (CPU 版本, 无 GPU 时使用)
|
||||
# 安装命令: pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
||||
torch>=2.0.0
|
||||
torchvision>=0.15.0
|
||||
|
||||
# 图像处理
|
||||
opencv-python>=4.8.0
|
||||
Pillow>=9.0.0
|
||||
numpy>=1.24.0
|
||||
scikit-image>=0.21.0
|
||||
|
||||
# 标注工具
|
||||
labelme>=6.0.0
|
||||
|
||||
# 进度条
|
||||
tqdm>=4.65.0
|
||||
Reference in New Issue
Block a user