Skip to content

Latest commit

 

History

History
82 lines (49 loc) · 1.83 KB

log.md

File metadata and controls

82 lines (49 loc) · 1.83 KB

训练记录

1.12

修改config.py文件中voc_data_dir='/home/csce/czwhhh/competition/rebar/VOC_GJ/'

先不修改min_size和max_size,感觉调大点会比较好,但是又怕显存不够。

voc_dataset.py 修改split='train'

VOC_BBOX_LABEL_NAMES只有person;

faster_rcnn_vgg16.py修改n_fg_class=1

vis_tool.py VOC_BBOX_LABEL_NAMES=('person')

trainer.py修改self.roi_cm = ConfusionMeter(2)

需要做的事:统计bbox的大小,重新修改anchor的scale大小,统计anchor


error:

FileNotFoundError: [Errno 2] No such file or directory:

'/home/csce/czwhhh/competition/rebar/VOC_GJ/Annotations/D0BF9992.xml'

去掉testDataloader,因为没有验证集。

第一次提交得分:0.32770400
第二次,使用原图大小2660*2000,0.96755500
第三次 anchor_scales=[8, 16, 32]变成[4, 8, 18] 0.96063200000
Conformable-CNN(Deprecated)
出现错误16:49:37] src/operator/././cudnn_algoreg-inl.h:112: Running performance tests to find the best convolution algorithm, this can take a while... (setting env variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable)
解决方案使用mxnet-1.0.0版本

1、使用FPN

2、加softnms 0.67176

https://www.smwenku.com/a/5b85ab9d2b71775d1cd39cc2/zh-cn/

# soft nms in faster_rcnn.py
det = np.concatenate((cls_bbox_l, prob_l.reshape(-1, 1)), axis=1)
keep = soft_nms(det)使用的是默认参数soft_nms(dets, sigma=0.5, Nt=0.3, threshold=0.001, method=1): score_threshold可以设置为0.7

sigma=0.5,Nt=0.1, threshold=0.7, 0.92662300000

sigma=0.5, Nt=0.5, threshold=0.7, method=2

3、调大roi数量。修改creator_tool.py中ProposalCreator默认参数

n_test_post_nms=350,  # 300
min_size=12  # 16
n_test_pre_nms=12000,
n_test_post_nms=2000,  # 300
min_size=10  # 16
anchor_scales=[4, 8, 16, 32]