{"id":7215,"date":"2024-04-11T11:01:01","date_gmt":"2024-04-11T03:01:01","guid":{"rendered":""},"modified":"2024-04-11T11:01:01","modified_gmt":"2024-04-11T03:01:01","slug":"\u76ee\u6807\u68c0\u6d4b\u4e4b\u6d4b\u8bd5\u65f6\u95f4\u589e\u5f3a\uff08TTA\uff09","status":"publish","type":"post","link":"https:\/\/mushiming.com\/7215.html","title":{"rendered":"\u76ee\u6807\u68c0\u6d4b\u4e4b\u6d4b\u8bd5\u65f6\u95f4\u589e\u5f3a\uff08TTA\uff09"},"content":{"rendered":"<p><svg xmlns=\"http:\/\/www.w3.org\/2000\/svg\" style=\"display: none;\"> \n <path stroke-linecap=\"round\" d=\"M5,0 0,2.5 5,5z\" id=\"raphael-marker-block\" style=\"-webkit-tap-highlight-color: rgba(0, 0, 0, 0);\"><\/path> \n<\/svg> <\/p>\n<h2>TTA<\/h2>\n<p>\u7b80\u5355\u8bf4\u660e\uff1a<br \/> augment\uff1a \u5bf9\u56fe\u50cf\u8fdb\u884cTTA\u64cd\u4f5c\uff1b<br \/> batch_augment\uff1a\u5bf9\u4e00\u6279\u56fe\u50cf\u8fdb\u884cTTA\u64cd\u4f5c\uff1b<br \/> deaugment_boxes: \u5bf9\u9884\u6d4b\u6846\u8fdb\u884cTTA\u64cd\u4f5c\uff0c\u8be5\u64cd\u4f5c\u4e0e\u8f93\u5165\u56fe\u50cf\u76f8\u53cd\u3002<\/p>\n<pre class=\"language-java prettyprint linenums\"><code class=\"prism language-python\"><span class=\"token keyword\">class<\/span> <span class=\"token class-name\">BaseWheatTTA<\/span><span class=\"token punctuation\">:<\/span>\n    <span class=\"token triple-quoted-string string\">\"\"\" author: @shonenkov \"\"\"<\/span>\n    image_size <span class=\"token operator\">=<\/span> <span class=\"token number\">512<\/span>\n\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">raise<\/span> NotImplementedError\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">batch_augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> images<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">raise<\/span> NotImplementedError\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">deaugment_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">raise<\/span> NotImplementedError\n\n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">TTAHorizontalFlip<\/span><span class=\"token punctuation\">(<\/span>BaseWheatTTA<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n    <span class=\"token triple-quoted-string string\">\"\"\" author: @shonenkov \"\"\"<\/span>\n\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> image<span class=\"token punctuation\">.<\/span>flip<span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">batch_augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> images<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> images<span class=\"token punctuation\">.<\/span>flip<span class=\"token punctuation\">(<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">deaugment_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> self<span class=\"token punctuation\">.<\/span>image_size <span class=\"token operator\">-<\/span> boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span>\n        <span class=\"token keyword\">return<\/span> boxes\n\n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">TTAVerticalFlip<\/span><span class=\"token punctuation\">(<\/span>BaseWheatTTA<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n    <span class=\"token triple-quoted-string string\">\"\"\" author: @shonenkov \"\"\"<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> image<span class=\"token punctuation\">.<\/span>flip<span class=\"token punctuation\">(<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">batch_augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> images<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> images<span class=\"token punctuation\">.<\/span>flip<span class=\"token punctuation\">(<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">)<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">deaugment_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> self<span class=\"token punctuation\">.<\/span>image_size <span class=\"token operator\">-<\/span> boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span>\n        <span class=\"token keyword\">return<\/span> boxes\n    \n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">TTARotate90<\/span><span class=\"token punctuation\">(<\/span>BaseWheatTTA<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n    <span class=\"token triple-quoted-string string\">\"\"\" author: @shonenkov \"\"\"<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> torch<span class=\"token punctuation\">.<\/span>rot90<span class=\"token punctuation\">(<\/span>image<span class=\"token punctuation\">,<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>\n\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">batch_augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> images<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">return<\/span> torch<span class=\"token punctuation\">.<\/span>rot90<span class=\"token punctuation\">(<\/span>images<span class=\"token punctuation\">,<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">3<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">deaugment_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        res_boxes <span class=\"token operator\">=<\/span> boxes<span class=\"token punctuation\">.<\/span>copy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>\n        res_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> self<span class=\"token punctuation\">.<\/span>image_size <span class=\"token operator\">-<\/span> boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span>\n        res_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span>\n        <span class=\"token keyword\">return<\/span> res_boxes\n\n<span class=\"token keyword\">class<\/span> <span class=\"token class-name\">TTACompose<\/span><span class=\"token punctuation\">(<\/span>BaseWheatTTA<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n    <span class=\"token triple-quoted-string string\">\"\"\" author: @shonenkov \"\"\"<\/span>\n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">__init__<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> transforms<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        self<span class=\"token punctuation\">.<\/span>transforms <span class=\"token operator\">=<\/span> transforms\n        \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">for<\/span> transform <span class=\"token keyword\">in<\/span> self<span class=\"token punctuation\">.<\/span>transforms<span class=\"token punctuation\">:<\/span>\n            image <span class=\"token operator\">=<\/span> transform<span class=\"token punctuation\">.<\/span>augment<span class=\"token punctuation\">(<\/span>image<span class=\"token punctuation\">)<\/span>\n        <span class=\"token keyword\">return<\/span> image\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">batch_augment<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> images<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">for<\/span> transform <span class=\"token keyword\">in<\/span> self<span class=\"token punctuation\">.<\/span>transforms<span class=\"token punctuation\">:<\/span>\n            images <span class=\"token operator\">=<\/span> transform<span class=\"token punctuation\">.<\/span>batch_augment<span class=\"token punctuation\">(<\/span>images<span class=\"token punctuation\">)<\/span>\n        <span class=\"token keyword\">return<\/span> images\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">prepare_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        result_boxes <span class=\"token operator\">=<\/span> boxes<span class=\"token punctuation\">.<\/span>copy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>\n        result_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> np<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">min<\/span><span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> axis<span class=\"token operator\">=<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span>\n        result_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> np<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">max<\/span><span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> axis<span class=\"token operator\">=<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span>\n        result_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> np<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">min<\/span><span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> axis<span class=\"token operator\">=<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span>\n        result_boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span> <span class=\"token operator\">=<\/span> np<span class=\"token punctuation\">.<\/span><span class=\"token builtin\">max<\/span><span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> axis<span class=\"token operator\">=<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">)<\/span>\n        <span class=\"token keyword\">return<\/span> result_boxes\n    \n    <span class=\"token keyword\">def<\/span> <span class=\"token function\">deaugment_boxes<\/span><span class=\"token punctuation\">(<\/span>self<span class=\"token punctuation\">,<\/span> boxes<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">:<\/span>\n        <span class=\"token keyword\">for<\/span> transform <span class=\"token keyword\">in<\/span> self<span class=\"token punctuation\">.<\/span>transforms<span class=\"token punctuation\">[<\/span><span class=\"token punctuation\">:<\/span><span class=\"token punctuation\">:<\/span><span class=\"token operator\">-<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">:<\/span>\n            boxes <span class=\"token operator\">=<\/span> transform<span class=\"token punctuation\">.<\/span>deaugment_boxes<span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">)<\/span>\n        <span class=\"token keyword\">return<\/span> self<span class=\"token punctuation\">.<\/span>prepare_boxes<span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">)<\/span>\n<\/code><\/pre>\n<h2>TTA\u7684\u7528\u6cd5<\/h2>\n<pre class=\"language-java prettyprint linenums\"><code class=\"prism language-python\"><span class=\"token comment\"># you can try own combinations:<\/span>\ntransform <span class=\"token operator\">=<\/span> TTACompose<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span>\n    TTARotate90<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span>\n    TTAVerticalFlip<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span>\n<span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span>\n\nfig<span class=\"token punctuation\">,<\/span> ax <span class=\"token operator\">=<\/span> plt<span class=\"token punctuation\">.<\/span>subplots<span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">3<\/span><span class=\"token punctuation\">,<\/span> figsize<span class=\"token operator\">=<\/span><span class=\"token punctuation\">(<\/span><span class=\"token number\">16<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">6<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>\n\nimage<span class=\"token punctuation\">,<\/span> image_id <span class=\"token operator\">=<\/span> dataset<span class=\"token punctuation\">[<\/span><span class=\"token number\">5<\/span><span class=\"token punctuation\">]<\/span>\n\nnumpy_image <span class=\"token operator\">=<\/span> image<span class=\"token punctuation\">.<\/span>permute<span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>cpu<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>numpy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>copy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>\n\nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>imshow<span class=\"token punctuation\">(<\/span>numpy_image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">;<\/span>\nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>set_title<span class=\"token punctuation\">(<\/span><span class=\"token string\">'original'<\/span><span class=\"token punctuation\">)<\/span>\n\ntta_image <span class=\"token operator\">=<\/span> transform<span class=\"token punctuation\">.<\/span>augment<span class=\"token punctuation\">(<\/span>image<span class=\"token punctuation\">)<\/span>\ntta_image_numpy <span class=\"token operator\">=<\/span> tta_image<span class=\"token punctuation\">.<\/span>permute<span class=\"token punctuation\">(<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">,<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>cpu<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>numpy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>copy<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span>\n\ndet <span class=\"token operator\">=<\/span> net<span class=\"token punctuation\">(<\/span>tta_image<span class=\"token punctuation\">.<\/span>unsqueeze<span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span><span class=\"token builtin\">float<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>cuda<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> torch<span class=\"token punctuation\">.<\/span>tensor<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span><span class=\"token builtin\">float<\/span><span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">.<\/span>cuda<span class=\"token punctuation\">(<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">)<\/span>\nboxes<span class=\"token punctuation\">,<\/span> scores <span class=\"token operator\">=<\/span> process_det<span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span> det<span class=\"token punctuation\">)<\/span>\n\n<span class=\"token keyword\">for<\/span> box <span class=\"token keyword\">in<\/span> boxes<span class=\"token punctuation\">:<\/span>\n    cv2<span class=\"token punctuation\">.<\/span>rectangle<span class=\"token punctuation\">(<\/span>tta_image_numpy<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>box<span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> box<span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>box<span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span>  box<span class=\"token punctuation\">[<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span>\n\nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>imshow<span class=\"token punctuation\">(<\/span>tta_image_numpy<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">;<\/span>\nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>set_title<span class=\"token punctuation\">(<\/span><span class=\"token string\">'tta'<\/span><span class=\"token punctuation\">)<\/span>\n    \nboxes <span class=\"token operator\">=<\/span> transform<span class=\"token punctuation\">.<\/span>deaugment_boxes<span class=\"token punctuation\">(<\/span>boxes<span class=\"token punctuation\">)<\/span>\n\n<span class=\"token keyword\">for<\/span> box <span class=\"token keyword\">in<\/span> boxes<span class=\"token punctuation\">:<\/span>\n    cv2<span class=\"token punctuation\">.<\/span>rectangle<span class=\"token punctuation\">(<\/span>numpy_image<span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>box<span class=\"token punctuation\">[<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span> box<span class=\"token punctuation\">[<\/span><span class=\"token number\">1<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span>box<span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">,<\/span>  box<span class=\"token punctuation\">[<\/span><span class=\"token number\">3<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token punctuation\">(<\/span><span class=\"token number\">0<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">1<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">0<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">,<\/span> <span class=\"token number\">2<\/span><span class=\"token punctuation\">)<\/span>\n    \nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>imshow<span class=\"token punctuation\">(<\/span>numpy_image<span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">;<\/span>\nax<span class=\"token punctuation\">[<\/span><span class=\"token number\">2<\/span><span class=\"token punctuation\">]<\/span><span class=\"token punctuation\">.<\/span>set_title<span class=\"token punctuation\">(<\/span><span class=\"token string\">'deaugment predictions'<\/span><span class=\"token punctuation\">)<\/span><span class=\"token punctuation\">;<\/span>\n<\/code><\/pre>\n<p><img decoding=\"async\" src=\"https:\/\/img-blog.csdnimg.cn\/20200910133303281.png\" =\"\" =\"\u5728\u8fd9\u91cc\u63d2\u5165\u56fe\u7247\u63cf\u8ff0\" alt=\"\u76ee\u6807\u68c0\u6d4b\u4e4b\u6d4b\u8bd5\u65f6\u95f4\u589e\u5f3a\uff08TTA\uff09 (https:\/\/mushiming.com\/)  \u7b2c1\u5f20\" title=\"\u76ee\u6807\u68c0\u6d4b\u4e4b\u6d4b\u8bd5\u65f6\u95f4\u589e\u5f3a\uff08TTA\uff09  \u7b2c1\u5f20-\u7a46\u4e16\u660e\u535a\u5ba2\" ><\/p>\n<h2>\u53c2\u8003<\/h2>\n<p><iframe class=\"wp-embedded-content\" sandbox=\"allow-scripts\" security=\"restricted\" title=\"[WBF over TTA][Single Model] EfficientDet\" src=\"https:\/\/www.kaggle.com\/embed\/shonenkov\/wbf-over-tta-single-model-efficientdet?kernelSessionId=35643411#?secret=C9MCRJW262\" data-secret=\"C9MCRJW262\" height=\"750\" frameborder=\"0\" scrolling=\"auto\"><\/iframe><\/p>\n","protected":false},"excerpt":{"rendered":"\u76ee\u6807\u68c0\u6d4b\u4e4b\u6d4b\u8bd5\u65f6\u95f4\u589e\u5f3a\uff08TTA\uff09TTA\u7b80\u5355\u8bf4\u660e\uff1aaugment\uff1a\u5bf9\u56fe\u50cf\u8fdb\u884cTTA\u64cd\u4f5c\uff1bbatch_augment\uff1a\u5bf9\u4e00\u6279\u56fe\u50cf\u8fdb\u884cTTA\u64cd\u4f5c\uff1bdeaugment...","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[],"tags":[],"_links":{"self":[{"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/posts\/7215"}],"collection":[{"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/comments?post=7215"}],"version-history":[{"count":0,"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/posts\/7215\/revisions"}],"wp:attachment":[{"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/media?parent=7215"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/categories?post=7215"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/mushiming.com\/wp-json\/wp\/v2\/tags?post=7215"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}