xbd1519722757

关于数据集设置的问题

11

作者您好！请问在跑这个代码之前，是否要自行将数据集中每张label的值修改成0、1？如果我想设置inference得到的结果颜色，请问是不是应该修改UV-SAM-main\mmpl\datasets\uv_seg_dataset.py中的以下代码？ ``` METAINFO = { 'classes': ('background', 'UV',), 'palette': [[120, 120, 120], [180, 120, 120],] } ``` 另外，在您公布的数据集中，标签图像background和UV的值似乎分别是[0,0,0]和[128,0,0]，为什么上述代码要设置为[120, 120, 120]和[180, 120, 120]？ 真诚盼望得到您的解答。祝您元宵节快乐！

大佬，我用一张2080，跑一个两千张512*512的数据集，batch size=16，怎么一个epoch要四十几分钟？同样的参数设置，在您分享的PSPNet和U-Net上都是几分钟一个epoch。请问可能是哪里出错了？