最近做了两个项目
一个是香烟检测任务,这个检测结果还可以,裱框挺准的,数据问题不再考虑之内
一个是天池大数据比赛-津南大数据啥安检的比赛
在调代码发现几个以前没注意的问题,
一、 代码使用基于github上面的mask-rcn修改得到的Faster-rcnn
这套代码很正规,不过也有不少工程上不足,改了不少工程问题。
mask-rcnn :https://github.com/matterport/Mask_RCNN
题主修改的 faster-rcnn: https://github.com/jiaowojiangdaye/faster-rcnn-tianchi-20190301
(题主的这个就是比赛用的,把输入数据形式都简化了,但也没写很详细的readme,谨慎使用)
常用参数列表:
#%% important args
Mode='train' #'train' or 'evaluate' or 'retrival'
Mode='evaluate'
#%% evaluate
if Mode == 'evaluate':
USING_NEGATIVE_IMG = True
USING_POSITIVE_IMG = False
NEGATIVE_MULT = 1
real_test = True # if true means that we load image without gt
init_with = "this" # imagenet, coco, or last this
EVA_LIMIT=10000
model_version = 'last0322_only_heads_0721_final'
weight_base = 'jinnan' # coco clothes
THIS_WEIGHT_PATH = 'logs/jinnan20190324T0119/mask_rcnn_jinnan_0721.h5'
if weight_base == 'coco':
NUM_CLASSES = 1000+1 # clothes has 80 classes
# Adjust