--- pengfei --- 2019年1月5日17:39:58
对 https://github.com/amdegroot/ssd.pytorch 的一些改造。使得在0.4.1版本的pytorch 也能获得同样的效果。
目前 train.py
文件留作备份,以pf_main.py
作为train的主文件
update
2019年1月9日15:48:03
引入了 keep_top_k
参数,用以控制每张图最多的有效bounding box数目(default 200)
说明: