-
项目结构
- model下面存各种分割的模型
- train.py用于训练
- predict.py用于训练之后结果的预测
- evaluate.py用于各种指标的计算
-
注意事项
-
模型修改在train.py中的create_model中修改model即可
-
训练前注意数据集的格式
-
root_path
-
train
-
image
-
mask
-
val
-
image
-
mask
-
mask中图片的像素要对应分割的类别数目，0默认为背景对应的类别，1，2，3等等为相应的目标区域对应的像素
-
使用某些模型进行分割是图片的大小需要保持一致，例如Unet++，可以通过pad_size.py对图片大小进行修改
-
-
Notifications
You must be signed in to change notification settings - Fork 0
TonFard/Segmentation
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
This is a image semantic segmentation refenence training scripts
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published