11 Commits

Author SHA1 Message Date
hanoch
221429a81b datasets.py: 768x1024, random padding, roi based scaling . still open issue the prediction over the random padding area should be omitted from calculation 2025-01-06 17:26:49 +02:00
hanoch
15755c31b9 dataset.py: align test() loader to train
scaling before mosaic with predefined filling_value according to img precision chnaged by scaling before mosaic. filling uint16 with image.mean()
detect.py : align to latest changes as test.py
test.py : support in large object size metrices stats_all_large
plots.py : plot detections with th>0.1 (not 0.25)
2024-11-04 10:08:35 +02:00
hanoch
c3276b3de9 Test/ train/detect :Scaling and normalization. 2024-09-15 09:20:39 +03:00
hanoch
5bb345f112 fix annotation txtx file creation 2024-08-07 18:03:48 +03:00
jpkoponen
072f76c72c
Fix a GPU memory leak in detect. No need to calculate gradients in inference. (#900)
Co-authored-by: JKO095 <juho-pekka.koponen@wartsila.com>
2022-10-07 23:50:04 +02:00
AlexeyAB84
591c5b0236 minor fix 2022-08-06 01:45:25 +03:00
AlexeyAB84
0a58f6d1cb minor fix 2022-08-02 00:45:17 +03:00
AlexeyAB84
2596994f39 minor fix 2022-07-21 20:37:15 +03:00
linghu8812
96390ed201
support onnx to tensorrt convert (#114) 2022-07-21 20:06:24 +03:00
AlexeyAB84
dfc68171c1 Added requirements.txt 2022-07-09 02:26:32 +03:00
Kin-Yiu, Wong
a1c6e04c7c
main code
code for inference
2022-07-06 23:23:27 +08:00