From 0455f35033a3ecd864b8d78b2e20c64dd8005a80 Mon Sep 17 00:00:00 2001
From: Vasilis Vryniotis <vvryniotis@fb.com>
Date: Thu, 14 Jan 2021 22:55:36 +0000
Subject: [PATCH] Reducing Resolution to 320 and anchor sizes to 8-128.

---
 ...t_fasterrcnn_mobilenet_v3_large_expect.pkl | Bin 4109 -> 4109 bytes
 ...sterrcnn_mobilenet_v3_large_fpn_expect.pkl | Bin 4109 -> 4109 bytes
 torchvision/models/detection/faster_rcnn.py   |   8 ++++----
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/test/expect/ModelTester.test_fasterrcnn_mobilenet_v3_large_expect.pkl b/test/expect/ModelTester.test_fasterrcnn_mobilenet_v3_large_expect.pkl
index 9a9207b73605db38072db7b2a7b5bfaf8a079b65..df07bea5c748b5db23a6aad08507e8ac3fba3e60 100644
GIT binary patch
literal 4109
zcmeHKZA?>V6u#xFSZqPGFa?}5RB<R@y(l;&-1nHWSyW&tf|<ihX$0-Ky|gf$L#il#
zgb}kS(U@t}xh!I`C7_wh=GQcuxJ1GIaa%|+OT<knM3*p(+1~TsYk@jU_HRGD$<6K4
zbDrls=iJY?*cuwfagmYS3tJ2q%Zc^&n%YK3dbQ{g(_EXJ7GmJmPT0I5>twD;)})k&
zQ-vyNv%@Wi*gX;@rd2k0YMpLfWxb=Sq1IVl>2cJ#rTWUoES(&hl2Wdv1UJnTH#p=l
zg`%@KCDHS)@~2J?FOQ}<nS4ezZ_dr;Gct2?Gk7_oJe2-xcQwe`mHRawIkLnWW1V3Q
zYn7+agQiv1?XlY>r_*8g)Jk=3-Rk;kN4=wZd9B@}lcN@t%2PeEzEn1pM^G83SmkiK
z<>(sfsOGtgBB^dez(h>2iHxj_9C=!e)>`e6V@s@I))~9x>DKUF@{GMTC~x%0aiwxR
zlXu&tdODHU6#1IqhO+ofo|k7TTC-HG1k_60!?cn-^6XOi^#(Dm*s6^@`p~DV)o>i{
z#gm@6L{2~9#HEiLAbPiuDPt!woz4KaSI1X#cy%4Tcpu<xm$x<V(j5BC!Tn{oUEHL%
z9X_`#BzEk!3p_({<{7Cx-1^f`&JG;}Q$-O}-`x(IzgPpC=C%RGtIE0w#y9qtz<{%k
zRFw^pfl)JT?3W-ZbtibwIpA>5UJwgEV)2%WVL^!-Vd)VwAoe^vNb`-*xY$p8hQ;{<
zqY~V?;wQ-GJ~&2@f3^JvS$f0>SI@YJ=UFSwH#2?iXNwwdBgjWT$baq!{bN1!uju{4
z@~a9%XuU3idIO^+Oz+O!hXU4rFluD=u|M>u^oM?pfO=RT_0WI8j>vrU-*UB!`SqTQ
zV*OSX&SCvzrpD3w63uUA^)=m}3E0n`q&L+3LN)(AHc!mw2fOIJm3;5H1!_DW&>zOp
zAI7mh&Rdxuof(Z|KN!dPV1MWj`$K={&s-5M{kSEstj$2)-kL0=cm75Yr&^vejQwIh
z_KW%Rq#EciXaT$TLkPRD59Ub*6(=#gqR&S#ex3W9V0=a2=M3NAeqeZ>v{21|i{&5q
zu0+k>t;Dx<3CM5P#4-GSS%vDSCE(}s{0{PamrIyAvzMU1n0#44%xCco?^w;N@ea~q
z+X&c?@ni#Fe}`-_Do$V+`Tf&Pfc*BUz~=3nE;8IUwHXY#El@4AK-lGd5Jmf$-}FdW
zKXP_i*`hcgrzB#n^b-ZU7(Qf6Q}uHL`riB5`6bsS64LG`>78Gy{ag(6^T{d~!MNtS
z84lTsRQ?*qfAZ!|DEMhhUMRgjzTTcJpx@}6AyRxzA~(Z%7<yJmw23@)-t$xJV*ZKd
z8aeQ73OU)NXZW~xEg<f>utSaSW%0tNeQaLGy|Y>TMAJJgzB^==8b6`T_x9Ok$CF*8
zsZJsz+x(;;*SE|zY=(vLF5-LoNLb<<5<2g7k&dWth7(*ag7`qx5aWy4-NGMt{RQ7Y
zHZuIxtWtuw@2^(!@kKL4=X4X>umm^5<5>MQ`crEB3Te7n!s4e#T3P(TtsXLe8i3BA
zg+aXtPwq#+%DEjxXE4F93B9EEn3*7sPwXIwF|OpBSiZs+Rem$$ubSJ-_=9>a#p_kQ
zW~TSx)<IfN3-O7)WKeIS@nou}rF@`zCct{g$9f7s0%~Jr0sVZlgeM9gRK5sGeUn-r
zs2-tu=trb_iXR1|U&P6v_``Zy%11w<s)uo+`W2a9^sn%n86WcxFXyShUOJBs=0}+i
zVw?~5gY{8g(ZhW7gZV|xbpqxu%HdhQQs1Q3NB>w4&j;2M**p*{^AKTVTNjB;j04-S
zk)FR&!1vSo$^ugX{g}o5W$&lOlf9qVXQ+i_{mc6)@S>XJ$Ab9Nd9U{W&kXQ-Nc!~g
zy8U#CfcxavLo{a&Tn~+odN*AwSxEYSy#&lUtk`pOgDr$UNpUZ>YVZ(PQ;uJ>#9&~;
z7QDg?W{>~&N}RwvPrDn>4qoO2D~-QPCMspnrR!u$_~jO?G=9Dlm2O8(s>JTGV5RYM
z!=>)n=F^WUsb{Cy8WFKj&6pj!D3FT_Z`__AE9hWwD>W!b%EP5L*Qr*(aIrNUSHF}K
J&SAci`wwCH34j0q

literal 4109
zcmeHKeN0nV6n{`!pwP<rc%ame5k=TkK3Zwr5PA1HHdol-EhsZnR!alKYTs*%aZW&;
z@?jhyQy4m;bBkLdGcHP&nK+}&#ASa>aT=!*r^dt}8<rS#%13t3y`_{Y?%(~vOI}`o
zJ?H$+@0|1Qd+!#S6(JlK7REhuL~+ra#c8dm++a&9v$!n_q_y@uq~%tQJGKO`lDTR*
zD5XS67}hv8+FWw5)$JgG3refpm3Ei9)M;B&RcS9Pb=%gv9L~}WnQB>)l2Q^%0$jw?
zQf`w&Scp2$?y$JuV}Gh;WyuuclVM0-Xv{Pkvy2((nG3UIRf&T9Yn7_x(1PY5w;Wbv
zjxxuZL+a#katW%F)eTmw!)~`(-Ib2@F7><4GMm#@wzSgfR?88oq8#a#HKMF7Q4t!u
zWsS}5lBZM<MHQne@*V5TeIlX)M5L!@WSZocDniX=ZaKQh9Ab{$A;*}NJLK5L3ZRd3
z%e*MZQ+k)x;UpbNVDuFM1!XNV8M93CRK_*U%asURx(3QM-7U`$<(XBM1%>9&u$`u~
zk;)*BgR^kL!Ra{7xWT3Q8p!hHq42RGrp6!6o;?3QJ%BM!|Lno4W-?+pUo__A3C3)s
zw`d+p8XNPtHR})3$QpoS@;cdv$@}2hbpY!z`{V4>v&cIK&KIpmexmhA-&=Vo`Z?<n
zdcL~WgADz8ym7b%1J2#nkYl?Shd(*1$LjN2@#obm@!zlP!o#PwVfWR3)cn?g9FA+n
z-~C*Kzd7D3+|>S%)BCgz|K8=m;a?v^zz25Vd;7+Rc$=aYpW|wA_~s^pRTTfi&7pYD
z$!^5#p%g>D^uY!M`5q45K!8_2no9Ab!LKMzA3RI(V6_G(f261JO+l>)`1jpeCjg&L
zVW1f5IWyOQ?;ZCD2~#CnpXz{w0P8bBC&x7?XXXI5l}VJpt(v3t4pqNO>)&)WQ~*7H
zJudLtv%U6mYHzEa;l;}+f7e`*=4*<q$EG_qc(<|^hiz-ZJkeKtw_Vt>J9zQ_<u9Y$
zhaCds&w6b@C^~l^1)Zy-@jE?p@%DX(QB34!if3K9wixi`gl4KY=GGU8*^|8fZ%}`=
z>QjnSx|UFTY-EVn9!~Aik*g_=xpjrwqa$<irA5~f==rud7=xa^_6W@IEQ&$j3e6nc
zchw=>c!FsB3XO&0KcCc4yydbafPK?lJ?d*u^~UG;;*U<dh*-Re`hP#ImEw~Rx{3dr
zUjGlM|Eb@|{M&*TUrXJEJ%b&>#RK<IR-=UCbq0LG=s|0VeQU;ce2Lf<7aB3x-$^oH
zu=gZN1Rp0@B$&tjntE#AyR?(W_ayT8&=LpQyVQf`n>?r|(SUo;$6>(nI+4U1aDHvS
z0RHn$oy5-5_&*HcIl%99NC@H~AB$)C3X*u5uP0F?e0t3COTPRJ8z^S^Ko8{O@jo?j
z*mzSy#RGM?H3IbLF{@Xi_4b}GqI_y?J&lL_V2AwB58z|`qL*KfE29U5c%6iQG<H(G
zppWSl@tMbcf==p09k=61K9S_tV~vT&#RDz)S7klEIk*mio{rmkst5YP;-McR?Jw&m
zPxV87rdRao1%J@X{O=>yADEAbJ;VJdQSuOe2i_AQ(Rx{bCE8z@AH%#4vwC=156qj*
z3-}DQ9<D1a2kN<=(1}1l)C2mV9yU*yH|RG|{Y)?K(+l-5y`%RB%pdL#p57l!Kg?@%
zzHq-u1WS}3^Z`Hg8|KIKfgR>qL^0!M80rH&dOrb%e1IXI<%4*Uu9t(2TAY8>lZ$Jn
zQ=A)f2m#(c)J2~M2OH;k?XOdNa6zX4_J}8O$TNHlIZqD=9k+|fJb7vtVs6p+9YZ51
zhCGk+wXG<5WHvtXw431BUW_PCKG}lWbOy9+s2d%5+JchF^V+Pvh~|u_@bY05hWFD?
zhf~^f$z}%5iQi9CIP7~rZLm3AWN(y*mOST`<o|j2pTG5LP|&@<yf~OVso*R$2f@X+
zr}W>nM1f)45xBz)jQ0N)){ToiLB5XtqXV}&flU5y@$pPwDkfrLU($h0{{0@$luq`v
z&)3BEu7OPceS@v;*l|HK#^jw=p;@Jx=k=JOnD6t2jW-;l*Y`<d;1IR!O5$Lv&2@M)
SfVt4DgxxQRQF7qVeE$W-6$Tdo

diff --git a/test/expect/ModelTester.test_fasterrcnn_mobilenet_v3_large_fpn_expect.pkl b/test/expect/ModelTester.test_fasterrcnn_mobilenet_v3_large_fpn_expect.pkl
index 9be3d021af619dcdb574cf36c78bcbffa44b5c8c..1fa8647e40e99accd145e7720d5884e0ce2a9fb7 100644
GIT binary patch
literal 4109
zcmeHKUrbY17{5^db}MwnF$82z#b9*h&u$x<A<%QS8mvOCC}^BmN<+I<+S^{3i6J2>
z1~O%YWqVLr$jAg;G%@o<6XVQWmTb$SfyFozebHnZH+g|&!H1pSIk(Vu!|ZVnoaE;A
z^gHMGec$i*opbJSIdy50l$9m@qZuWW<cWCwfgWFZhe!38!yg14BuCnt(E3u}Q>0!c
zrK~lbA+(1M`JzgyR}Ha==C*D%5RB^EBEI(SK(M1t^>sx<k+z;Hy`n2CYt3X6qHLt6
z)2F0~DSAgR<WYBuKlMs_Yc3m8DOXrzxymA2trn|QRx(<3>|bxVTglu$oT4gOJDf)6
z3TN7JC7Zocjw|{BuQwD7`n+l&)D_h?MLK*DU&lKEuc}vG+Td1RRuzL=$!X1CFhNhd
zFBnyF{VXDXEQ)QRuFi!Zj7dRQs$~|$EB;JphpL!%IMbXf`jtFqdcU$_&=2@mRVCl8
zyvE_9-cW=wQXt^{$q8AiZPp59rNAoGu!?|HJixJ5sY;1kdA-|Xb~!V%E{v*w1X3gk
zk84TciX}sWaOH~x^4J|Ddr`!g@XK#j46H5lu*^ez9bnmgd~5UcYPRR#dC{`VUwYZG
zW$qxR#bvj!ht{4xU$gxDe>i~NQ(x~=huId3=S6z2UrKL&PC3Y$f6;qEc<sk?|3xx+
zz(`M>y+Ve@i>T<EwZ4u1_C;OoT*hT`|3sL~neu6`e2`8a*hU+tB6MiHh4m|G(~XCu
zamq?pZJ8&%a-P;N;C<lP(!aF>zj`Q#Joi5#J1dUS(`RSN_qPW*9_+w_9sXuBW;A~?
zi{}YBb7T*}_*GkuGW>&d;=udd|H<)g9xw2KFYxlHz$@T*y>b(a*T}{<vv`YW-Kl=o
zH`9s00PBnVo*Kg3-|wX}1a{!T9>Wv)7HIkAaeL!b0k<PR!~_3`2mUZ0{xJX9_+2~v
zS+?sKzc!w);48xULB4m3X6)eiY3XI2U*nBQ#wYC8IiI2rKChEMZ+CD$E!z)sKJTyl
zjz|}aYh33Hw0X9S+YvA9&<FgX5B_3$34W}cpU34gg@4=HFol2Q1AoXz_y>Q`8~j0U
z^vzn|M&bX*bEgUXpWnKM0uOuw4}5lwNc6Lvb<Dnz+0$s(NC_3=8~AwWDfo(TzUDvj
z(+7Q9>CXcn(rd@|Fn^Vr-OBBc%k$}BT|TW|>!x!iH?!wa)CcrHeOzlO<nfHT#|h#A
zzZfs-*Ujq%{EPhNiO>gpL7x^nyFT0g!%YXN|FV(jFPz}@DE+d(2J?S)hIu|Be&n-;
zLJ!0bz8C8S{GeWJ6!AQqewQMi=F*c~5AY8?Q18$B?vc#0Onc$JRC=tZk>P`%JB%;z
zQ_u4g^$WZrdgz-Cv@cRZ3-{I1vC#&tp0{cB+`{V(`XL|aC-gUK`rCN^&=dK`>Otsf
z<?%jPH^}(AL&n?>7=QIlFNtoN7_vjZ=1+g(c&K;8k9tS`;1l^{{iBa~Mg58O20PXh
z?8d9PHHdE_AlJw08TyI&M;YJOHy_r?Os^UHoGDBZPe=Pb9#8-7RIX3er7aZt^zYus
z>j`>+FX#pS#CmAp>jC_xJi1B1@7Sn;kN;$Fl;zVvO?%(q`Xe9kCD!jjt)8&{u-`Ok
z`&l!OAN&hE@NeMjLDV1YM+xG6G@Z@#27mi`{D=pAtWWGOB7g89^Z`F&f1Bs?t1fNe
z^ZR$#GJNc}*uSvf7SZs?UJ5=Yr;kwZ0e$)^SH2B>*2$&zu8|ro|4yDi_(1(3U(_G$
zg0D`_7xEGHSIg@UeDy^nR$t(2BZXdKzm)66{#B{%2dI}8iu3Bs)xU=8*<}XL(&tsK
zL>JDh9$zHN?nVw$_aBa^<RdGg|CSyR-2S|CS1S8#!sBwL;I(k4jK8!PVMu7nH_YVO
z@!wv>2{W&-(>Z>2@--(JDgIPXM5<s{*CmnIms>JYJl}~(tJ$4)2_*hakc<@18?L&G
z?UrE?NmI+^%*fcJjVRH*xiA(NZ?xEZ4<iQ6og?0C9<JKbSDFWyUCwmee%YLK3FF1s
FzX79IU)KNt

literal 4109
zcmeHKZ){Ul6o2jhcXTTQLD(E~1F}UNqinc^TDN;m7wCp%3L9inyRNQ<X5DMMg`mjD
z*kH+!2~m^6V3B0N6eE!kB@q(91QCf7h^P_#0MQRKE}B4eeDMCxecQU4p&$LQm%O~5
zo_qfN&iUQ<uEQP|uPDjM%0DiXlA*YQZm+M+v$W9_b``d4@!MFMvT@SYu76oowyV1Q
zx&+~&A+XgGQuXd|Ko}M}Tf;to$lwfm8d`n+MrYX590~-TZDxZSm!DskBn(1grK`!K
z#!C}}%^z@uUzC3u)P%Zpv1VCuQOPrA^RhDY@}d<*rD|ebocPz>(yAs^_v*rG@>;ve
zKHDDOrKX67u1huSak~S4zsDW+1)4*K7lMtRpr`S9pF3<&QwwU;8DZ5}qo&m*3O9aN
zgU26I)4d`i?^qU9f##<1Bup_$6c;Zum#Q<pN%qFDnz7a%Z=c<%&ax+Ts<XSjaQ|3X
zol~RECHEnBASe)dT)Owh7Gy3lm#t9eNni6dUzzZgwTFB?5mvKn)CH}sLWeymIr*=v
zU-)#2g4;1=<FXXvq{c0s0FdI@Sjt2btp^7!)A#=e1GM@2$LqqqqQ$r$)#k-hwV8Ng
zW4RM;p2XQ`;la>xW<Hh5ORCSa&h<H5$_1Ki<)*Xytlwp}uw%Z{Y<N*If9T!J&8I5)
z!GTuZxxQY=g}kxh2NoG96S7{&CQZ&H8U7A*_f^1u#>OYvgQ2@Da%-pXKg!rsJ>(C1
z_#4w-xnyI|8&38x#4A4K(c*2R{G|PP&ECXe9~sCb`-k2tkzXp04AhGFIedHZE+H52
zHP^d^Y$X}?kY#)m)dPE3kBLJM8G7W8_!6IOBHt=aUrYKQD=${Szb}#|@Uv1qz(eAr
zC;XI~w=KZ$d;190<7qfTdc(=fLN;;8hzI$%qZ>55ItZ_Giw`lyb=cD7%H_j12T2e7
zOR76Wyn2z}UC}?tqa>pr5|1*%;~VE3j`*c{yE)>c9>haEGCuIC)bK(-As1+}l_P%S
zX9J`7oz+6V#V%OhE$?!b^Wn~=+&Rm@rGAr;tEqnAhx&ovA9G#&GacuLX71pxrMQH?
zP{Xf`^heK+uwml}Yq7QPUl+9qy@{j$5|18+`GX$#L67`_uk44OOZ+P}{Oh@?_f>1$
zm3i#k>0ZWG9wofwymXAui>wd*w(`+PZ$<Mb^Lg6Et4vYeAw1>0juQU27k<v#le;XA
z6GnbyZ!h&r&KK-+c=P^je(y#Kcbr(o2Ul;>=B-kjw|dGCcw;_*x9k`Ag??G79@#(e
zPx49T2R>Q(8`b6PtI$ri+Z8YH-9q`vd4&Boj(L%IU>@`&OFl!E^AEk8f0{=N@)3BJ
zkIkFpTL<y&m-n8t-cOvvZ}+sY`x(0_zV6;7ir?R7Bz)V077l#-`<kf#z`M(}i34xo
z_2r3!40sK$&Lw?Peo_VWyV}+;^b7e*zJO1gso$6v$@eO4{{VmWgs=1md~?P6Ds4V$
zX+7`(KL{Ve5AYrJq24jPB|gAQPxZ71vpLrH_pPG%z)SKqN91ecsP|&gUZFSAyeoPA
zD>3i8&U{eeoR!aIyw{13n19R<>cjkC9^|~NqIr>ce5>J6j{U%b{sMowUleNmL%${e
zB!6p(zs>tCn!hIUC+9s?!!L*G*%91I^$cBmi@$O%(F#5q^9mUFcxPcZTXO8pivF5A
z?4`G_GVuAMGo>QmT*?>tT?_q5{k?zT0Dth)40il%4_kb?n)Ij-c%VMuf%%pD2lnU7
z4DYM`A7=etE<UDkKk9w8VygEQeNx$&`9Hj`qTgNkJ@(-`{S+Uw;+n=`8i(jP06+JA
z{p^h#ac06jH9ympym5ZEd4eHvcCxVr|3=B>21FP-+_X+F!s2$=b$E=QDWeZ9CMYId
zu_w$}>*#OctVx!iim&5n>)2yXtf%N*YO<$(fy)%0=u0}*Q#78FJvj_hdZIlp)>AZY
zIO<MZBgP3&S~nc_#Ka}qitM<j$JgTEjce?=K>&lRCQUwt9gf<{`&tC39rgsAeuYhf
Ig7tFk-{vS_X8-^I

diff --git a/torchvision/models/detection/faster_rcnn.py b/torchvision/models/detection/faster_rcnn.py
index 1529298b1bc..5e90dc58763 100644
--- a/torchvision/models/detection/faster_rcnn.py
+++ b/torchvision/models/detection/faster_rcnn.py
@@ -372,7 +372,7 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True,
 
 
 def fasterrcnn_mobilenet_v3_large(pretrained=False, progress=True, num_classes=91, pretrained_backbone=True,
-                                  trainable_backbone_layers=None, min_size=320, max_size=640, **kwargs):
+                                  trainable_backbone_layers=None, min_size=320, max_size=320, **kwargs):
     """
     Constructs a Faster R-CNN model with a MobileNetV3-Large backbone. It works similarly
     to Faster R-CNN with ResNet-50 FPN backbone. See `fasterrcnn_resnet50_fpn` for more details.
@@ -402,7 +402,7 @@ def fasterrcnn_mobilenet_v3_large(pretrained=False, progress=True, num_classes=9
     backbone = mobilenet_backbone("mobilenet_v3_large", pretrained_backbone, False,
                                   trainable_layers=trainable_backbone_layers)
 
-    anchor_sizes = ((16, 32, 64, 128, 256), )
+    anchor_sizes = ((8, 16, 32, 64, 128, ), )
     aspect_ratios = ((0.5, 1.0, 2.0), )
 
     model = FasterRCNN(backbone, num_classes, rpn_anchor_generator=AnchorGenerator(anchor_sizes, aspect_ratios),
@@ -414,7 +414,7 @@ def fasterrcnn_mobilenet_v3_large(pretrained=False, progress=True, num_classes=9
 
 
 def fasterrcnn_mobilenet_v3_large_fpn(pretrained=False, progress=True, num_classes=91, pretrained_backbone=True,
-                                      trainable_backbone_layers=None, min_size=320, max_size=640, **kwargs):
+                                      trainable_backbone_layers=None, min_size=320, max_size=320, **kwargs):
     """
     Constructs a Faster R-CNN model with a MobileNetV3-Large FPN backbone. It works similarly
     to Faster R-CNN with ResNet-50 FPN backbone. See `fasterrcnn_resnet50_fpn` for more details.
@@ -444,7 +444,7 @@ def fasterrcnn_mobilenet_v3_large_fpn(pretrained=False, progress=True, num_class
     backbone = mobilenet_backbone("mobilenet_v3_large", pretrained_backbone, True,
                                   trainable_layers=trainable_backbone_layers)
 
-    anchor_sizes = ((16, 32, 64, 128, 256, ), ) * 3
+    anchor_sizes = ((8, 16, 32, 64, 128, ), ) * 3
     aspect_ratios = ((0.5, 1.0, 2.0),) * len(anchor_sizes)
 
     model = FasterRCNN(backbone, num_classes, rpn_anchor_generator=AnchorGenerator(anchor_sizes, aspect_ratios),