Commit Graph
500 Commits
Author SHA1 Message Date
xdobro23 8473e4eb62 exporter: 16bit single-channel network support 2024-06-27 13:56:38 +00:00
xdobro23 9e42d05e01 disabled AMP check (not supported anyway) 2024-06-01 14:01:16 +00:00
xdobro23 d92dce3a46 augmentations for 16bit grayscaled 2024-04-29 20:57:53 +02:00
xdobro23 8a7dc71f0e added logging 2024-04-29 16:12:09 +02:00
xdobro23 d15113ed0a disabled augmentations 2024-04-29 16:11:57 +02:00
Clea Parcerisas 82b97d5d4c changed data/loaders and engine/predictor so they accept 1 channel images for predictions 2023-08-24 14:42:29 +02:00
Clea Parcerisas 95e2c2424a changed the imgsize on enginge/prediction to meet the grayscale for prediction 2023-08-24 11:09:35 +02:00
Clea Parcerisas ba4c33dc53 changed the imgsize on enginge/validator to meet the grayscale for prediction 2023-08-24 10:54:18 +02:00
Clea Parcerisas d221ace5a3 changed the imgsize on warmup to meet the grayscale for prediction 2023-08-24 10:51:30 +02:00
Clea Parcerisas 833d864689 added changes to use grayscale images during training 2023-08-21 10:51:22 +02:00
Clea Parcerisas e8dcc30754 changed the weights for the fitness function 2023-08-21 09:46:04 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3acead7e79 Add TensorBoard graph for model visualization (#4464)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-21 01:22:29 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Yonghye Kwon
87ce15d383 ultralytics 8.0.158 add benchmarks to coverage (#4432)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
2023-08-20 20:52:30 +02:00
LaughingGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
8f79ce45c1 Fix conversion ops using clone and copy (#4438)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-20 13:15:37 +02:00
Glenn JocherandGitHub 162e4035eb Docker-python fix for get_ubuntu_version() (#4430) 2023-08-17 19:32:12 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
9d27e7ada4 New ASSETS and trackers GMC cleanup (#4425)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-17 18:19:05 +02:00
aaba14e6b2 ultralytics 8.0.157 fix unzip_file() bug (#4407)
Co-authored-by: Nadim Bou Alwan <64587372+nadinator@users.noreply.github.com>
2023-08-17 13:19:09 +02:00
Glenn JocherandGitHub 8d7490f060 ultralytics 8.0.156 fix PIL *.png EXIF bug (#4406) 2023-08-16 19:31:35 +02:00
fb1ae9bfad Fix LoadStreams final frame bug (#4387)
Co-authored-by: Nadim Bou Alwan <64587372+nadinator@users.noreply.github.com>
2023-08-16 19:00:03 +02:00
ZouJiu1andGitHub 17e6b9c270 Correct scipy.optimize.linear_sum_assignment usage (#4390) 2023-08-16 16:17:42 +02:00
9a0555eca4 ultralytics 8.0.155 allow imgsz and batch resume changes (#4366)
Co-authored-by: Mostafa Nemati <58460889+monemati@users.noreply.github.com>
Co-authored-by: Eduard Voiculescu <eduardvoiculescu95@gmail.com>
2023-08-15 22:02:23 +02:00
LaughingGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
60cad0c592 Cleanup tracker and remove unused functions (#4374)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-15 20:18:39 +02:00
Glenn JocherandGitHub 4093c1fd64 Add data/utils.py to tests (#4363) 2023-08-15 01:25:48 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
834f94f899 Fix PIL Image exif_size() function (#4355)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-14 17:48:54 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
c940d29d4f Omit ultralytics/utils/callbacks from coverage (#4345)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-14 03:25:51 +02:00
Glenn JocherandGitHub d47718c367 ultralytics 8.0.154 add freeze training argument (#4329) 2023-08-14 00:18:32 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
9f6d48d3cf Improve tests coverage and speed (#4340)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-13 22:24:01 +02:00
d704507217 Clean and bump dvc callback: settings, stacked images (#4343)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-13 21:50:35 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
9366062af2 Remove unused code (#4327)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-13 13:10:52 +02:00
Glenn JocherandGitHub 822608986c ultralytics 8.0.153 YOLO Tasks Cleanup (#4314) 2023-08-12 02:30:57 +02:00
Glenn JocherandGitHub 39395aedc8 Improved Classify dataset splits and introspection (#4312) 2023-08-11 20:39:50 +02:00
Gustavo RibeiroGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
645c715350 Implementing Resource Release Mechanism for Stream Loading (#4285)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-11 15:28:32 +02:00
LaughingandGitHub b73f9b13df Fix incorrect device selection when resuming training (#4301) 2023-08-11 14:32:39 +02:00
Glenn JocherandGitHub 28893d9d2d ultralytics 8.0.152 bump torch>=1.8.0 (#4299) 2023-08-11 00:49:55 +02:00
Glenn JocherandGitHub 6267d6a609 Fix torch.Tensor inference results plotting (#4297) 2023-08-10 20:43:40 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Kayzwer
c9be1f3cce ultralytics 8.0.151 add DOTAv2.yaml for OBB training (#4258)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kayzwer <68285002+Kayzwer@users.noreply.github.com>
2023-08-10 00:55:36 +02:00
Simon LévesqueGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
a76af55533 Fixes Results.tojson when tracking (#4270)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-10 00:19:28 +02:00
LaughingGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
29ace4982d Add "assert len(labels)" catch (#4257)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-10 00:15:14 +02:00
AndyGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
dde89c744c Refactor val code into new self.match_predictions() method (#4265)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-09 19:03:46 +02:00
LaughingGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
e9c9b82c42 Fix save_hybrid (#4245)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-09 18:51:21 +02:00
Glenn JocherGitHubMLBoy_DaisukeMajimapre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Louis Lac
7dfdb63cde ultralytics 8.0.150 new Apple *.mlpackage export format (#4043)
Co-authored-by: MLBoy_DaisukeMajima <rockyshikoku@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Louis Lac <lac.louis5@gmail.com>
2023-08-08 17:00:42 +02:00
Glenn JocherandGitHub f5fc807fa7 Accept model=yolov5*u.yaml arguments (#4230) 2023-08-08 14:09:11 +02:00
pre-commit-ci[bot]GitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>Glenn Jocher
7dcdca3bf4 Replace type(1) == int with isinstance(1, int) (#4217)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-08 01:53:47 +02:00
Glenn JocherandGitHub ff5fa57415 Improved Docs models Usage examples (#4214) 2023-08-07 20:57:35 +02:00
Glenn JocherandGitHub 9a2c0691e3 Scope scipy import and only plot first 1M labels (#4203) 2023-08-07 03:57:16 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>AdiEcho
7565210484 ultralytics 8.0.149 add Open Images V7 training (#4178)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: AdiEcho <30563671+AdiEcho@users.noreply.github.com>
2023-08-07 01:19:51 +02:00
Glenn JocherGitHubpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
c751c7f88a Enable Google Drive file downloads (#4196)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-06 20:11:14 +02:00
Glenn JocherandGitHub eb80ba9ce0 Update check_version() for inequality support (#4182) 2023-08-05 18:23:51 +02:00
Glenn JocherandGitHub 038558cfab ultralytics 8.0.148 fix SettingsManager empty YAML bug (#4180) 2023-08-05 16:13:29 +02:00
730af8390b Fix error with exporting quantized saved model when labels cache not present (#4174)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2023-08-05 15:35:14 +02:00