{"id":875,"date":"2020-12-18T14:20:19","date_gmt":"2020-12-18T06:20:19","guid":{"rendered":"http:\/\/www.yatenglg.cn\/?p=875"},"modified":"2022-10-18T16:37:12","modified_gmt":"2022-10-18T08:37:12","slug":"tensorflow-object-detection-api-with-tensorflow-1-%e7%9b%ae%e6%a0%87%e6%a3%80%e6%b5%8b","status":"publish","type":"post","link":"http:\/\/www.yatenglg.cn\/blog\/?p=875","title":{"rendered":"TensorFlow Object Detection API with TensorFlow 1 \u76ee\u6807\u68c0\u6d4b"},"content":{"rendered":"

Object Detection API with TensorFlow 1<\/a><\/p>\n

\u73af\u5883\u4f9d\u8d56<\/strong><\/p>\n

    \n
  1. tensorflow==1.15.0<\/li>\n
  2. cuda10.0 > 10.0<\/li>\n
  3. cudnn > 7.6<\/li>\n
  4. protoc > 3.0<\/li>\n<\/ol>\n

    \u4e0b\u8f7d<\/strong>\uff1a<\/p>\n

    \n
    \n
    git clone https:\/\/github.com\/tensorflow\/models.git<\/pre>\n<\/div>\n<\/div>\n
    \u5b89\u88c5<\/strong>\uff1a\uff08pip\u7248\u672c\u8f83\u4f4e\u4f1a\u9020\u6210\u5b89\u88c5\u5931\u8d25\uff09<\/p>\n
    cd<\/span> models\/research\n# Compile protos.<\/span>\nprotoc object_detection\/protos\/*<\/span>.proto --python_out=.\n# Install TensorFlow Object Detection API.<\/span>\ncp object_detection\/packages\/tf1\/setup.py .<\/span>\npython -m pip install --use-feature=2020-resolver .<\/span><\/pre>\n
    \u6d4b\u8bd5<\/strong>\uff1a<\/p>\n
    python object_detection\/builders\/model_builder_tf1_test.py<\/pre>\n
    \u7ed3\u6784<\/strong>\uff1a<\/p>\n
    object detection api\u76f8\u5173\u6587\u4ef6\u5b58\u653e\u5728\/research\/object_detection\/<\/p>\n
    \u8bad\u7ec3<\/strong>\uff1a\uff08\u4f7f\u7528\u4e0b\u8f7d\u7684\u6587\u4ef6\u8fdb\u884c\u8bad\u7ec3\uff09<\/p>\n
    \u4ee5\u4f7f\u7528voc\u6570\u636e\u96c6\u8bad\u7ec3faster_rcnn_inception_resnet_v2\u4e3a\u4f8b\uff1a<\/p>\n
    \u9700\u51c6\u5907config\u914d\u7f6e\u6587\u4ef6\uff0c\u6570\u636e\u96c6\u8f6ctfrecord\u3002<\/p>\n
    1.\u65b0\u5efa\u9879\u76ee\u6587\u4ef6\u5939voc<\/p>\n
    2.\u5c06\u4e0b\u8f7d\u7684voc2012\u6570\u636e\u96c6\u653e\u5165voc\u6587\u4ef6\u5939\u4e2d\uff0c\u89e3\u538b\u751f\u6210VOCdevkit\u6587\u4ef6\u5939<\/p>\n
    3.\u6570\u636e\u96c6\u51c6\u5907\uff08research\/object_detection\u76ee\u5f55\u4e0b\uff09<\/p>\n
    python dataset_tools\/create_pascal_tf_record.py --data_dir voc\/VOCdevkit\/ --year=VOC2012 <\/span> --set=train --output_path=voc\/pascal_train.record <\/span><\/pre>\n
    python dataset_tools\/create_pascal_tf_record.py --data_dir voc\/VOCdevkit\/ --year=VOC2012 <\/span>--set=val --output_path=voc\/pascal_val.record<\/pre>\n
    \u751f\u6210\u8bad\u7ec3\u548c\u6d4b\u8bd5\u6570\u636e\uff1a pascal_train.record<\/span> \uff0c pascal_val.record<\/p>\n
    \u590d\u5236data\/pascal_label_map.pbtxt\u5230voc\u6587\u4ef6\u5939\u4e0b\uff0c\u6b64\u4e3a\u6570\u636e\u6807\u7b7e\u6587\u4ef6<\/p>\n
    4. \u9884\u8bad\u7ec3\u6a21\u578b\u4e0b\u8f7d<\/p>\n
    \u53bbhttps:\/\/github.com\/tensorflow\/models\/blob\/master\/research\/object_detection\/g3doc\/tf1_detection_zoo.md\u4e0b\u8f7dfaster_rcnn_inception_resnet_v2_atrous_coco<\/a>\u6587\u4ef6\u3002<\/p>\n
    \u590d\u5236\u7684voc\u6587\u4ef6\u5939\u540e\u89e3\u538b\u3002<\/p>\n
    \u751f\u6210faster_rcnn_inception_resnet_v2_atrous_coco_2018_01_28\u6587\u4ef6\u5939\uff08\u5185\u5b58\u653e\u4e86\u4ee5\u53ca\u8bad\u7ec3\u597d\u7684\u9884\u8bad\u7ec3\u6a21\u578b\uff09<\/p>\n
    5.\u914d\u7f6e\u6587\u4ef6config<\/p>\n
    \u4ecesamples\/configs\/ \u590d\u5236 faster_rcnn_inception_resnet_v2_atrous_coco.config \u5230voc\u76ee\u5f55\u4e0b<\/p>\n
    \u4fee\u6539\u914d\u7f6e\u6587\u4ef6\u4e2d\uff08\u51715\u5904\u4fee\u6539\uff09\uff1a<\/p>\n
    fine_tune_checkpoint: <\/span>\"<\/span>voc\/faster_rcnn_inception_resnet_v2_atrous_coco_2018_01_28\/model.ckpt<\/span>\"<\/span>#\u9884\u8bad\u7ec3\u6a21\u578b\u5b58\u653e\u5904<\/span><\/pre>\n
    train_input_reader: {\ntf_record_input_reader {\ninput_path: <\/span>\"<\/span>voc\/pascal_train.record<\/span>\"<\/span>#\u8bad\u7ec3\u6570\u636e\u96c6\n}\nlabel_map_path: <\/span>\"<\/span>voc\/pascal_label_map.pbtxt<\/span>\"<\/span>#\u8bad\u7ec3\u6570\u636e\u96c6\u6807\u7b7e\u6587\u4ef6\n}<\/span><\/pre>\n
    eval_input_reader: {\ntf_record_input_reader {\ninput_path: <\/span>\"<\/span>voc\/pascal_val.record<\/span>\"<\/span>#\u6d4b\u8bd5\u6570\u636e\u96c6\n}\nlabel_map_path: <\/span>\"<\/span>voc\/pascal_label_map.pbtxt<\/span>\"<\/span>#\u6d4b\u8bd5\u6570\u636e\u96c6\u6807\u7b7e\u6587\u4ef6\nshuffle: <\/span>false<\/span>\nnum_readers: <\/span>1<\/span>\n}<\/span><\/span><\/pre>\n
    6.\u8bad\u7ec3<\/p>\n
    \n
    python model_main.py --model_dir voc\/train_dir\/ --pipeline_config_path voc\/faster_rcnn_inception_resnet_v2_atrous_voc.config<\/pre>\n
    \u6216<\/p>\n
    python legacy\/train.py --train_dir voc\/train_dir\/ --pipeline_config_path voc\/faster_rcnn_inception_resnet_v2_atrous_voc.config<\/pre>\n
    \u5f00\u59cb\u8bad\u7ec3<\/p>\n
    \"\"<\/p>\n<\/div>\n","protected":false},"excerpt":{"rendered":"
    Object Detection API with TensorFlow 1 \u73af\u5883\u4f9d\u8d56 tensorflow=…<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":[],"categories":[8,5],"tags":[],"_links":{"self":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/875"}],"collection":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=875"}],"version-history":[{"count":3,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/875\/revisions"}],"predecessor-version":[{"id":1162,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=\/wp\/v2\/posts\/875\/revisions\/1162"}],"wp:attachment":[{"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=875"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=875"},{"taxonomy":"post_tag","embeddable":true,"href":"http:\/\/www.yatenglg.cn\/blog\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=875"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}