Mengkai Yan

Results 9 issues of Mengkai Yan

def nms(boxes, scores, overlap=0.5, top_k=200): keep = scores.new(scores.size(0)).zero_().long() if boxes.numel() == 0: **_return keep_** x1 = boxes[:, 0] y1 = boxes[:, 1] x2 = boxes[:, 2] y2 = boxes[:, 3]...

Hi: I want to apply this code to the TUM dataset. What changes should I make?thank you very much!!!

Why is the input size of the trained sample [511,511] but the input size of the test sample is : inp_height = new_height | 127 inp_width = new_width | 127...

ImportError: /root/.local/lib/python3.7/site-packages/cpools-0.0.0-py3.7-linux-x86_64.egg/top_pool.cpython-37m-x86_64-linux-gnu.so: undefined symbol: _ZN2at6detail20DynamicCUDAInterface10set_deviceE

您好,我想问一下,如果程序因为bug中断的话,train函数里的线程会自动停止吗? 担心程序在调试过程中会开启很多线程而无法停止。

mtcnn test very very slow,please check if you have time

if i train my own data,where should i modify

RuntimeError: Error(s) in loading state_dict for DummyModule: Unexpected key(s) in state_dict: "module.pre.0.bn.num_batches_tracked", "module.pre.1.bn1.num_batches_tracked", "module.pre.1.bn2.num_batches_tracked", "module.pre.1.skip.1.num_batches_tracked", "module.kps.0.up1.0.bn1.num_batches_tracked", "module.kps.0.up1.0.bn2.num_batches_tracked", "module.kps.0.up1.1.bn1.num_batches_tracked", "module.kps.0.up1.1.bn2.num_batches_tracked", "module.kps.0.low1.0.bn1.num_batches_tracked", "module.kps.0.low1.0.bn2.num_batches_tracked", "module.kps.0.low1.0.skip.1.num_batches_tracked", "module.kps.0.low1.1.bn1.num_batches_tracked", "module.kps.0.low1.1.bn2.num_batches_tracked", "module.kps.0.low2.up1.0.bn1.num_batches_tracked", "module.kps.0.low2.up1.0.bn2.num_batches_tracked", "module.kps.0.low2.up1.1.bn1.num_batches_tracked", "module.kps.0.low2.up1.1.bn2.num_batches_tracked", "module.kps.0.low2.low1.0.bn1.num_batches_tracked", "module.kps.0.low2.low1.0.bn2.num_batches_tracked",...