yolov5 / models

Commit History

Parameterize ONNX `--opset-version` (#3154)
9ab561d
unverified

Cristi Fati commited on

rename class autoShape -> AutoShape (#3173)
be86c21
unverified

developer0hye commited on

Add `--include torchscript onnx coreml` argument (#3137)
d9b4e6b
unverified

Cristi Fati glenn-jocher commited on

Add --optimize argument (#3093)
525f4f8
unverified

glenn-jocher commited on

Update P5 + P6 model ensembling (#3082)
251aeaf
unverified

glenn-jocher commited on

Update export.py with --train mode argument (#3066)
e97d129
unverified

glenn-jocher commited on

Do not optimize CoreML TorchScript model (#3055)
3ef3a95
unverified

glenn-jocher commited on

Fix ONNX export using --grid --simplify --dynamic simultaneously (#2982)
b292837
unverified

whoam glenn-jocher commited on

Add support for FP16 (half) to export.py (#3010)
e2a80c6
unverified

Hodovo glenn-jocher commited on

YOLOv5 AWS Inferentia Inplace compatibility updates (#2953)
41f5cc5
unverified

jluntamazon glenn-jocher commited on

New Colors() class (#2963)
57812df
unverified

glenn-jocher commited on

Add yolov5/ to sys.path() for *.py subdir exec (#2949)
4200674
unverified

glenn-jocher commited on

Improved yolo.py profiling (#2940)
4890499
unverified

glenn-jocher commited on

Add verbose option to pytorch hub models (#2926)
c0d3f80
unverified

Nanobit glenn-jocher commited on

Add file_size() function (#2911)
1b1ab4c
unverified

glenn-jocher commited on

Update export.py for 2 dry runs (#2910)
646386f
unverified

glenn-jocher commited on

Update export.py (#2909)
a2a514d
unverified

glenn-jocher commited on

Default optimize_for_mobile() on TorchScript models (#2908)
28db237
unverified

glenn-jocher commited on

Update yolo.py (#2899)
cdb678f
unverified

glenn-jocher commited on

Implement yaml.safe_load() (#2876)
f7bc685
unverified

glenn-jocher commited on

Detection cropping+saving feature addition for detect.py and PyTorch Hub (#2827)
c949fc8
unverified

burhan glenn-jocher commited on

Fix ONNX dynamic axes export support with onnx simplifier, make onnx simplifier optional (#2856)
1df8c6c
unverified

Tim Stokman Tim glenn-jocher commited on

PyTorch Hub cv2 .save() .show() bug fix (#2831)
c15e25c
unverified

glenn-jocher commited on

ONNX Simplifier (#2815)
1f3e482
unverified

glenn-jocher commited on

YOLOv5 v5.0 Release patch 1 (#2764)
0f395b3
unverified

glenn-jocher commited on

torch.cuda.amp bug fix (#2750)
b5de52c
unverified

glenn-jocher commited on

autocast enable=torch.cuda.is_available() (#2748)
fca5e2a
unverified

glenn-jocher commited on

Add Hub results.pandas() method (#2725)
c03d590
unverified

glenn-jocher commited on

Updated filename attributes for YOLOv5 Hub BytesIO (#2718)
ec8979f
unverified

glenn-jocher commited on

Updated filename attributes for YOLOv5 Hub results (#2708)
74276d5
unverified

glenn-jocher commited on

autoShape forward im = np.asarray(im) # to numpy (#2689)
17300a4
unverified

glenn-jocher commited on

PyTorch Hub model.save() increment as runs/hub/exp (#2684)
2af059c
unverified

glenn-jocher commited on

Add TransformerLayer, TransformerBlock, C3TR modules (#2333)
1148e2e
unverified

Ding Yiwei DingYiwei glenn-jocher commited on

PyTorch Hub amp.autocast() inference (#2641)
2bf34f5
unverified

glenn-jocher commited on

Update Detections() self.n comment (#2620)
005d7a8
unverified

glenn-jocher commited on

Update detections() self.t = tuple() (#2617)
a57f23d
unverified

glenn-jocher commited on

Update Detections() times=None (#2570)
1c132a1
unverified

glenn-jocher commited on

Improved W&B integration (#2125)
e8fc97a
unverified

Ayush Chaurasia glenn-jocher commited on

autoShape() speed profiling update (#2460)
f813f6d
unverified

glenn-jocher commited on

Add autoShape() speed profiling (#2459)
569757e
unverified

glenn-jocher commited on

GPU export options (#2297)
7a0a81f
unverified

Jan Hajek Jan Hajek glenn-jocher commited on

Anchor override (#2350)
a3ecf0f
unverified

glenn-jocher commited on

EMA bug fix 2 (#2330)
fab5085
unverified

glenn-jocher commited on

remove TTA 1 pixel offset (#2325)
fd96810
unverified

glenn-jocher commited on

Update yolo.py (#2120)
cbd55da
unverified

oleg Oleg Boiko glenn-jocher train255 ab-101 Transigent Nanobit Daniel Khromov VdLMV Jaap van de Loosdrecht Yann Defretin Aditya Lohia Shivam Swanrkar xiaowo1996 Iden Craven Jan Hajek Jan Hajek commited on

Dynamic ONNX engine generation (#2208)
95aefea
unverified

Aditya Lohia Shivam Swanrkar glenn-jocher commited on

Update inference default to multi_label=False (#2252)
c09964c
unverified

glenn-jocher commited on

Unified hub and detect.py box and labels plotting (#2243)
5f42643
unverified

Yann Defretin commited on

Improved hubconf.py CI tests (#2251)
db28ce6
unverified

glenn-jocher commited on

YOLOv5 Hub URL inference bug fix (#2250)
9d87307
unverified

glenn-jocher commited on