DeepGrabCut-PyTorch

DeepGrabCut-PyTorch copied to clipboard

several questions

1

Hi,thanks for you provide this project,and several questions make me confused: 1.while training,what is the GT images?The bounding-box annotation images?Or the mask annotation images?Or no GT images? 2.The input images...

zhenmafan7

GPU memory consumption is too large, is there any optimization method, is it because the network framework uses resnet101?

psw13583112

while I trying to run the train.py, the followed error occured： Traceback (most recent call last): File "/home/wq/Code/DeepGrabCut/train.py", line 147, in loss = class_balanced_cross_entropy_loss(output, gts, size_average=True, batch_average=True) File "/home/wq/Code/DeepGrabCut/layers/loss.py", line...

zhenmafan7