class weights of cityscapes

fregu856 / deeplabv3

PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.

http://www.fregu856.com/

MIT License

766 stars 179 forks source link

class weights of cityscapes #4

Open Chenfeng1271 opened 5 years ago

Chenfeng1271 commented 5 years ago

I don't find the weight file 'class_weights.pkl',can you tell me have to get it

fregu856 commented 5 years ago

Hi,

You'll have to run utils/preprocess_data.py.

// Fredrik

Chenfeng1271 commented 5 years ago

Thank u, but I get some new troubles when I train this code directly. It occur 'RuntimeError('cuda runtime error (77) : an illegal memory access was encountered '. So I add 'torch.cuda.set_device(1)' on the top, but here's a new one 'RuntimeError('cuda runtime error (4) : unspecified launch failure at ' These all happen at batch_losses.append(loss_value) in train.py. I try to skip it and sth correspond to it, but the same thing happens in loss.backward(). Could you give me some advice?

ShristiDasBiswas commented 2 years ago

Hi,

You'll have to run utils/preprocess_data.py.

// Fredrik Hi,

The eval code for generating IOU values is missing. Kindly push that code too.