search

wuyang98 / CoSOD

7 stars 1 forks source link

代码运行 #2

Closed Wuhuddy closed 10 months ago

Wuhuddy commented 10 months ago

你好，运行coseg_test.py时出现以下报错： image 请问如何解决，以及训练时代码运行顺序应该是怎样的呢？

wuyang98 commented 10 months ago

你好，感谢你对我们工作的关注。在训练阶段，需要先训练vqvae，然后固定住vqvae训练pixelcnn，最后进行整个模型的训练，训练过程中vqvae权重为冻结状态，pixelcnn权重为未冻结状态。整理好的代码和readme将在不久后更新。

Wuhuddy commented 9 months ago

image 你好，请问dateset.py的这块代码中contour是什么数据？我在训练过程中报错找不到这个目录。

wuyang98 commented 9 months ago

你好，contour是二值化标签的边缘信息，可以使用draw_contour.py为数据集中的二值化标签提取边缘信息，并用于训练。同时，也可以对代码进行修改，以不使用边缘信息，默认是使用的。

Wuhuddy commented 9 months ago

好的我明白了，感谢回复