- 23 3月, 2021 1 次提交
-
-
由 Ayush Chaurasia 提交于
* Init Commit * new wandb integration * Update * Use data_dict in test * Updates * Update: scope of log_img * Update: scope of log_img * Update * Update: Fix logging conditions * Add tqdm bar, support for .txt dataset format * Improve Result table Logger * Init Commit * new wandb integration * Update * Use data_dict in test * Updates * Update: scope of log_img * Update: scope of log_img * Update * Update: Fix logging conditions * Add tqdm bar, support for .txt dataset format * Improve Result table Logger * Add dataset creation in training script * Change scope: self.wandb_run * Add wandb-artifact:// natively you can now use --resume with wandb run links * Add suuport for logging dataset while training * Cleanup * Fix: Merge conflict * Fix: CI tests * Automatically use wandb config * Fix: Resume * Fix: CI * Enhance: Using val_table * More resume enhancement * FIX : CI * Add alias * Get useful opt config data * train.py cleanup * Cleanup train.py * more cleanup * Cleanup| CI fix * Reformat using PEP8 * FIX:CI * rebase * remove uneccesary changes * remove uneccesary changes * remove uneccesary changes * remove unecessary chage from test.py * FIX: resume from local checkpoint * FIX:resume * FIX:resume * Reformat * Performance improvement * Fix local resume * Fix local resume * FIX:CI * Fix: CI * Imporve image logging * (:(:Redo CI tests:):) * Remember epochs when resuming * Remember epochs when resuming * Update DDP location Potential fix for #2405 * PEP8 reformat * 0.25 confidence threshold * reset train.py plots syntax to previous * reset epochs completed syntax to previous * reset space to previous * remove brackets * reset comment to previous * Update: is_coco check, remove unused code * Remove redundant print statement * Remove wandb imports * remove dsviz logger from test.py * Remove redundant change from test.py * remove redundant changes from train.py * reformat and improvements * Fix typo * Add tqdm tqdm progress when scanning files, naming improvements Co-authored-by:Glenn Jocher <glenn.jocher@ultralytics.com>
-
- 15 3月, 2021 4 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
* PyTorch Hub models default to CUDA:0 if available * device as string bug fix
-
-
由 Yann Defretin 提交于
* Be able to create dataset from annotated images only Add the ability to create a dataset/splits only with images that have an annotation file, i.e a .txt file, associated to it. As we talked about this, the absence of a txt file could mean two things: * either the image wasn't yet labelled by someone, * either there is no object to detect. When it's easy to create small datasets, when you have to create datasets with thousands of images (and more coming), it's hard to track where you at and you don't want to wait to have all of them annotated before starting to train. Which means some images would lack txt files and annotations, resulting in label inconsistency as you say in #2313. By adding the annotated_only argument to the function, people could create, if they want to, datasets/splits only with images that were labelled, for sure. * Cleanup and update print() Co-authored-by:
-
- 14 3月, 2021 4 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
* Add autoShape() speed profiling * Update common.py * Create README.md * Update hubconf.py * cleanuip
-
- 13 3月, 2021 4 次提交
-
-
由 Glenn Jocher 提交于
curl preferred over wget for slightly better cross platform compatibility (i.e. out of the box macos compatible).
-
由 Glenn Jocher 提交于
* labels.png class names * fontsize=10
-
由 Glenn Jocher 提交于
* Update test.py --task train val study * update argparser --task
-
由 Glenn Jocher 提交于
* Integer printout * test.py 'Labels' * Update train.py
-
- 10 3月, 2021 3 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
由 Kartikeya Sharma 提交于
* added argoverse-download ability * bugfix * add support for Argoverse dataset * Refactored code * renamed to argoverse-HD * unzip -q and YOLOv5 small cleanup items * add image counts Co-authored-by:
Kartikeya Sharma <kartikes@trinity.vision.cs.cmu.edu> Co-authored-by:
Kartikeya Sharma <kartikes@trinity-0-32.eth> Co-authored-by:
-
- 08 3月, 2021 2 次提交
-
-
由 Glenn Jocher 提交于
* GCP sudo docker * cleanup
-
由 Glenn Jocher 提交于
-
- 07 3月, 2021 2 次提交
-
-
由 Glenn Jocher 提交于
-
由 Jan Hajek 提交于
* option for skip last layer and cuda export support * added parameter device * fix import * cleanup 1 * cleanup 2 * opt-in grid --grid will export with grid computation, default export will skip grid (same as current) * default --device cpu GPU export causes ONNX and CoreML errors. Co-authored-by:
Jan Hajek <jan.hajek@gmail.com> Co-authored-by:
-
- 06 3月, 2021 4 次提交
-
-
由 Yonghye Kwon 提交于
image weights compatible faster random index generator v2 for mosaic augmentation
-
由 Glenn Jocher 提交于
Set HOME environment variable per Binder requirements. https://github.com/binder-examples/minimal-dockerfile
-
由 Glenn Jocher 提交于
-
由 Yonghye Kwon 提交于
* faster random index generator for mosaic augementation We don't need to access list to generate random index It makes augmentation slower. * Update datasets.py Co-authored-by:
-
- 04 3月, 2021 1 次提交
-
-
由 Glenn Jocher 提交于
* Resume with custom anchors fix * Update train.py
-
- 03 3月, 2021 4 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
由 Ryan Avery 提交于
-
由 Glenn Jocher 提交于
-
- 02 3月, 2021 1 次提交
-
-
由 Glenn Jocher 提交于
* EMA bug fix 2 * update
-
- 01 3月, 2021 1 次提交
-
-
由 Glenn Jocher 提交于
-
- 28 2月, 2021 3 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
-
- 27 2月, 2021 1 次提交
-
-
由 oleg 提交于
* Avoid mutable state in Detect * LoadImages() pathlib update (#2140) * Unique *.cache filenames fix (#2134) * fix #2121 * Update test.py * Update train.py * Update autoanchor.py * Update datasets.py * Update log_dataset.py * Update datasets.py Co-authored-by:
Jaap van de Loosdrecht <jaap.van.de.loosdrecht@nhlstenden.com> Co-authored-by:
Shivam Swanrkar <ss8464@nyu.edu> Co-authored-by:
xiaowo1996 <429740343@qq.com> * Update README.md * Amazon AWS EC2 startup and re-startup scripts (#2185) * Amazon AWS EC2 startup and re-startup scripts * Create resume.py * cleanup * Amazon AWS EC2 startup and re-startup scripts (#2282) * Update train.py (#2290) * Update train.py * Update train.py * Update train.py * Update train.py * Create train.py * Improved model+EMA checkpointing (#2292) * Enhanced model+EMA checkpointing * update * bug fix * bug fix 2 * always save optimizer * ema half * remove model.float() * model half * carry ema/model in fp32 * rm model.float() * both to float always * cleanup * cleanup * Improved model+EMA checkpointing 2 (#2295) * Fix labels being missed when image extension appears twice in filename (#2300) * W&B entity support (#2298) * W&B entity support * shorten wandb_entity to entity Co-authored-by:
Oleg Boiko <oboiko@chegg.com> Co-authored-by:
train255 <thanhdd.it@gmail.com> Co-authored-by:
ab-101 <56578530+ab-101@users.noreply.github.com> Co-authored-by:
Transigent <wbdsmith@optusnet.com.au> Co-authored-by:
NanoCode012 <kevinvong@rocketmail.com> Co-authored-by:
Daniel Khromov <danielkhromov@gmail.com> Co-authored-by:
VdLMV <jaap@vdlmv.nl> Co-authored-by:
Yann Defretin <kinoute@gmail.com> Co-authored-by:
Aditya Lohia <64709773+aditya-dl@users.noreply.github.com> Co-authored-by:
Iden Craven <iden.craven@gmail.com> Co-authored-by:
Jan Hajek <toretak@users.noreply.github.com> Co-authored-by:
-
- 26 2月, 2021 2 次提交
-
-
由 Jan Hajek 提交于
* W&B entity support * shorten wandb_entity to entity Co-authored-by:
-
由 Iden Craven 提交于
-
- 25 2月, 2021 3 次提交
-
-
由 Glenn Jocher 提交于
-
由 Glenn Jocher 提交于
* Enhanced model+EMA checkpointing * update * bug fix * bug fix 2 * always save optimizer * ema half * remove model.float() * model half * carry ema/model in fp32 * rm model.float() * both to float always * cleanup * cleanup
-
由 Glenn Jocher 提交于
* Update train.py * Update train.py * Update train.py * Update train.py * Create train.py
-
