OHEM

[hchoi71]

Hello, Yude.

Thanks for sharing this great work!

I have one question about table 1. You mentioned that you reported results in table 1 with the training set. Then, it seems OHEM process should be involved with train_SEAM.py. Is that correct? Does your repo include OHEM process? How can I use OHEM in your code?

Thanks

[YudeWang]

Hi @hchoi71 ,
OHEM means the topk selection here.

SEAM/train_SEAM.py

Line 137 in c556016

loss_ecr1 = torch.mean(torch.topk(tensor_ecr1.view(ns,-1), k=(int)(21*hs*ws*0.2), dim=-1)[0])

[hchoi71]

Oh I see. Thanks @YudeWang