PixelLink-with-pytorch

PixelLink-with-pytorch copied to clipboard

where to download best accuaracy model? 哪里下载最佳模型？

nissansz

Problem in Loss Func. Use ICDAR2015 to train to model, but this problem emerged.

2

File "/workspace/PixelLink-with-pytorch/main.py", line 144, in main() File "/workspace/PixelLink-with-pytorch/main.py", line 133, in main train(config.epoch, iteration, dataloader, my_net, optimizer, optimizer2, device) File "/workspace/PixelLink-with-pytorch/main.py", line 73, in train pixel_pos_weights) File "/workspace/PixelLink-with-pytorch/criterion.py", line 46,...

leonrenlang

RuntimeError: each element in list of batch should be of equal size

2

自己的数据集，修改好训练，报这个错，有大神遇到吗？ raise RuntimeError('each element in list of batch should be of equal size') RuntimeError: each element in list of batch should be of equal size

liuguoxionglang

Pretrained model

2

Sorry but where can I download the pretrained model, or which version do I need?

JinShchen

我跑了ICDAR2015数据集，loss一般在05左右，降不下去了，训练集上测试效果几乎没效果，请问是还没有收敛吗？

ligaoyuan-potato

iteration 12: pixel_loss: nan, link_loss: nan, total loss: nan, time: 3.7550010681152344

4

我改了数据加载这里，看了下没啥问题，为啥会爆炸呢？请指教一下 reade_label() with codecs.open(os.path.join(dir, self.labels_name[i-1]), encoding="utf-8_sig") as file: data = file.readlines() temp_str = '' for l in data: temp_str += l.strip('\r\n') temp_json = json.loads(temp_str) tmp = {} tmp["coor"] = []...

ligaoyuan-potato

Problem in training.

1

File "main.py", line 138, in main() File "main.py", line 128, in main train(config.epoch, iteration, dataloader, my_net, optimizer, optimizer2, device) File "main.py", line 59, in train for i_batch, sample in enumerate(dataloader):...

vdsprakash

instead of image_height, image_width was written

dragonsan17

cuda out of memory

1

when I train this model, it happened this situation. And I have a RTX 1080 Ti with 12GB memory.

curry8

Is there any update?How can i test?

4

hello, thanks for your codes and am interested in it.. As you said this version is still under developing, is there any new updates? I didn't find the functions for...

Redaimao