HRNet / HRNet-Image-Classification

Train the HRNet model on ImageNet
https://jingdongwang2017.github.io/Projects/HRNet/
MIT License
960 stars 215 forks source link

about self.class_weights #24

Open yuanry16 opened 4 years ago

yuanry16 commented 4 years ago

HI! from ~\lib\datasets\cityscapes.py i can see: self.class_weights = torch.FloatTensor([0.8373, 0.918, 0.866, 1.0345, 1.0166, 0.9969, 0.9754, 1.0489, 0.8786, 1.0023, 0.9539, 0.9843, 1.1116, 0.9037, 1.0865, 1.0955, 1.0865, 1.1529, 1.0507]).cuda()

What do you mean?