Simplify autoshape() post-process (#1653)

* Simplify autoshape() post-process

* cleanup

* cleanup

requirements.txt

@@ -9,8 +9,8 @@ Pillow
 PyYAML>=5.3
 scipy>=1.4.1
 tensorboard>=2.2
-torch>=1.6.0
-torchvision>=0.7.0
+torch>=1.7.0
+torchvision>=0.8.1
 tqdm>=4.41.0
 
 # logging -------------------------------------
@@ -26,5 +26,5 @@ pandas
 # scikit-learn==0.19.2  # for coreml quantization
 
 # extras --------------------------------------
-# thop  # FLOPS computation
-# pycocotools>=2.0  # COCO mAP
+thop  # FLOPS computation
+pycocotools>=2.0  # COCO mAP