Commit Graph

Select branches

Hide Pull Requests

G0_raw_decode_output

G0_yolov5s6_raw

G10_decoupled_head_light_FGEM_safe

G11_cls_head_enhance_Lightweight

G12_cls_head_enhance_and_ECA

G13_decoupled_head_ECA

G1_backbone

G2_neck

G2_neck_2

G3_head

G4_cls_loss_gain

G5_focal_loss

G6_decoupled_head

G7_RawHead_add_ECA

G8_decoupled_head_light

G9_decoupled_head_light_FGEM

M0_yolov5_raw

master

Mono Color

• fbf41e0913 Add train.run() method (#3700) Glenn Jocher 2021-06-20 15:06:58 +02:00
• c1af67dcd4 Add torch DP warning (#3698) Glenn Jocher 2021-06-19 19:50:46 +02:00
• b3e2f4e08d Eliminate total_batch_size variable (#3697) Glenn Jocher 2021-06-19 19:14:59 +02:00
• fad27c0046 Update DDP for torch.distributed.run with gloo backend (#3680) Glenn Jocher 2021-06-19 16:30:25 +02:00
• 5bab9a28e4 Reformat (#3694) Glenn Jocher 2021-06-19 14:21:28 +02:00
• bfb2276b1d Slightly modify CLI execution (#3687) lb-desupervisedandLewis Belcher 2021-06-19 12:06:59 +02:00
• bf209f6fe9 Skip HSV augmentation when hyperparameters are [0, 0, 0] (#3686) Mai Thanh MinhandGlenn Jocher 2021-06-19 16:51:21 +07:00
• 814806c61d Update cache check (#3691) Glenn Jocher 2021-06-19 11:22:09 +02:00
• 463628a4d8 Update README.md (#3684) Glenn Jocher 2021-06-18 17:12:42 +02:00
• 2729761458 EMA changes for pre-model's batch_size (#3681) ZouJiu1andGlenn Jocher 2021-06-18 22:26:52 +08:00
• f527704cd3 Cache v0.3: improved corrupt image/label reporting (#3676) Glenn Jocher 2021-06-18 10:21:47 +02:00
• 2296f1546f Update WORLD_SIZE and RANK retrieval (#3670) Glenn Jocher 2021-06-17 23:24:30 +02:00
• 045d5d8629 Update TensorBoard (#3669) Glenn Jocher 2021-06-17 22:12:42 +02:00
• fa201f968e Update train(hyp, *args) to accept hyp file or dict (#3668) Glenn Jocher 2021-06-17 22:03:25 +02:00
• 6d6e2ca65f Update train.py (#3667) Glenn Jocher 2021-06-17 21:32:39 +02:00
• ac34834563 Update download_weights.sh comment (#3662) Glenn Jocher 2021-06-17 16:21:10 +02:00
• 2754adad46 Remove /weights directory (#3659) Glenn Jocher 2021-06-17 14:44:12 +02:00
• 9b6dba6207 Update dataset_stats() to list of dicts (#3657) Glenn Jocher 2021-06-17 13:59:52 +02:00
• df7706d802 Update README.md (#3650) SpongeBab 2021-06-17 18:37:53 +08:00
• fa29125f18 is_coco list fix (#3646) Mai Thanh Minh 2021-06-17 03:56:16 +07:00
• 6187edcb53 Update CONTRIBUTING.md (#3647) Glenn Jocher 2021-06-16 19:57:11 +02:00
• 3ce0db89b0 Update CONTRIBUTING.md (#3645) Glenn Jocher 2021-06-16 19:48:30 +02:00
• d808855f77 Assert non-premature end of JPEG images (#3638) xiaowk5516andGlenn Jocher 2021-06-16 19:31:26 +08:00
• bb79e13d52 RUN pip install --no-cache -U torch torchvision (#3637) Glenn Jocher 2021-06-16 12:35:33 +02:00
• 6c0e1d9fd7 Update verify_image_label() (#3635) Glenn Jocher 2021-06-16 11:12:15 +02:00
• 65f81bfefa Add **/*.torchscript.pt (#3634) Glenn Jocher 2021-06-16 10:49:50 +02:00
• de56813ba8 FROM nvcr.io/nvidia/pytorch:21.05-py3 (#3633) Glenn Jocher 2021-06-16 10:40:31 +02:00
• 30e4c4f092 Update README.md (#3624) Glenn Jocher 2021-06-15 15:44:10 +02:00
• 7d3686a686 Update check_file() (#3622) Glenn Jocher 2021-06-15 13:21:04 +02:00
• 4c5d9bff80 Fix incorrect end epoch comment (#3612) Wei Quan 2021-06-15 05:24:56 -04:00
• daab682b06 Suppress wandb images size mismatch warning (#3611) Ayush ChaurasiaandGlenn Jocher 2021-06-14 22:24:58 +05:30
• 239a11c197 Download COCO and VOC by default (#3608) Glenn Jocher 2021-06-14 13:58:54 +02:00
• 6062319ec3 Update datasets.py (#3591) masood azharandGlenn Jocher 2021-06-14 03:28:18 -07:00
• f8adee156d merge develop Glenn Jocher 2021-06-14 10:45:15 +02:00
• 3133607560 Simplify README.md (#3530) Kalen MichaelandGlenn Jocher 2021-06-13 02:37:20 +02:00
• 88b1945241 Delete __init__.py (#3596) Glenn Jocher 2021-06-12 15:21:37 +02:00
• 7a565f130a Update dataset_stats() (#3593) Glenn Jocher 2021-06-12 13:26:41 +02:00
• 53d4fc2e26 W&B: Allow changed in config variable #3588 Ayush Chaurasia 2021-06-12 02:48:57 +05:30
• 4984cf54be train.py GPU memory fix (#3590) Glenn Jocher 2021-06-11 20:24:03 +02:00
• e8c5237403 ConfusionMatrix normalize=True fix (#3587) Glenn Jocher 2021-06-11 11:46:05 +02:00
• ec2da4a82c Add ConfusionMatrix normalize=True flag (#3586) Glenn Jocher 2021-06-11 11:37:08 +02:00
• 46e1fdfbc6 Update stale.yml (#3585) Glenn Jocher 2021-06-11 11:23:39 +02:00
• 5c32bd3080 Created using Colaboratory Glenn Jocher 2021-06-10 16:01:34 +02:00
• 53ed872c28 Update export.py, yolo.py sys.path.append() (#3579) Glenn Jocher 2021-06-10 15:35:22 +02:00
• 095197bd4a Ignore Seaborn plot warnings (#3576) Glenn Jocher 2021-06-10 13:51:29 +02:00
• 4695ca8314 Refactoring cleanup (#3565) Glenn Jocher 2021-06-09 22:50:27 +02:00
• 0e5cfdbea7 Refactor models/export.py arguments (#3564) Glenn Jocher 2021-06-09 22:43:46 +02:00
• 66cf5c28c1 Refactor detect.py arguments (#3559) Glenn Jocher 2021-06-09 22:19:34 +02:00
• a9553c04a7 Refactor test.py arguments (#3558) Glenn Jocher 2021-06-09 21:36:10 +02:00
• c6deb73a89 Bump pip from 18.1 to 19.2 in /utils/google_app_engine (#3561) dependabot[bot] 2021-06-09 21:14:09 +02:00
• 4ef9261870 Remove redundant speed/study half argument (#3557) Glenn Jocher 2021-06-09 16:30:01 +02:00
• 5948f20a3d Update test.py profiling (#3555) Glenn Jocher 2021-06-09 16:25:17 +02:00
• 8b5086c21b Multi-GPU default to single device 0 (#3554) Glenn Jocher 2021-06-09 15:31:47 +02:00
• 63157d214d Remove is_coco argument from test() (#3553) Glenn Jocher 2021-06-09 15:09:51 +02:00
• 958ab92dc1 Remove opt from create_dataloader()` (#3552) Glenn Jocher 2021-06-09 13:14:56 +02:00
• 0cfc5b2c18 Update README.md (#3550) Glenn Jocher 2021-06-09 11:49:25 +02:00
• f8ec71e1c2 Update check_python(minimum=3.6.2) (#3548) Glenn Jocher 2021-06-09 11:34:15 +02:00
• ef0b5c9d29 On-demand pycocotools pip install (#3547) Glenn Jocher 2021-06-09 11:22:21 +02:00
• 1b5edb6f8e Update dataset_stats() for HUB (#3536) Glenn Jocher 2021-06-09 10:56:11 +02:00
• b6fdd2e5e5 Create dataset_stats() for HUB Glenn Jocher 2021-06-08 23:09:45 +02:00
• ac8691e208 Created using Colaboratory Glenn Jocher 2021-06-08 19:01:52 +02:00
• 78cf488556 Created using Colaboratory Glenn Jocher 2021-06-08 18:54:33 +02:00
• c6b51f4189 Update FP16 --half argument for test.py and detect.py (#3532) Glenn Jocher 2021-06-08 18:47:13 +02:00
• 8d52c1c5c5 Update datasets.py (#3531) Glenn Jocher 2021-06-08 18:36:40 +02:00
• ed651873ff Merge branch 'develop' Glenn Jocher 2021-06-08 18:14:18 +02:00
• 28bff22df8 Use multi-threading in cache_labels (#3505) Dean MarkandGlenn Jocher 2021-06-08 19:00:21 +03:00
• f3c3d2ce5d Merge develop branch into master (#3518) +111 people 2021-06-08 10:22:10 +02:00
• c058a61e3b Update greetings.yml Glenn Jocher 2021-06-08 10:11:06 +02:00
• abb2a96e91 Update README.md (#3512) Glenn Jocher 2021-06-08 00:39:02 +02:00
• d986145b9a Reduce test.py results spacing (#3511) Glenn Jocher 2021-06-07 23:21:34 +02:00
• eede7dc48c Update tutorial.ipynb (#3510) Glenn Jocher 2021-06-07 22:52:16 +02:00
• 8d1ddc93c7 Earlier assert for cpu and half option (#3508) Yonghye Kwon 2021-06-08 01:56:41 +09:00
• 3f03acb3db check_requirements() exclude opencv-python (#3507) Glenn Jocher 2021-06-07 15:38:51 +02:00
• 90b7895d65 check_requirements() exclude opencv-python (#3495) Glenn Jocher 2021-06-07 10:03:47 +02:00
• a1c3572bc9 Explicit model.eval() call if opt.train=False (#3475) Yonghye KwonandGlenn Jocher 2021-06-07 01:39:36 +09:00
• cf4f95bc5f Stale github.actor bug fix (#3483) Glenn Jocher 2021-06-06 18:06:24 +02:00
• 3597d280ee Update stale.yml HUB URL (#3468) Glenn Jocher 2021-06-04 23:49:43 +02:00
• 739451da5a Add additional links/resources to stale.yml message (#3467) Glenn Jocher 2021-06-04 23:45:24 +02:00
• b31229ae89 Revert FP16 test.py and detect.py inference to FP32 default (#3423) edificewangandGlenn Jocher 2021-06-05 04:28:34 +08:00
• 044daafd9d Add output_names argument for ONNX export with dynamic axes (#3456) 3 people 2021-06-05 00:01:58 +04:00
• 317f2ccc9d Improved check_requirements() offline-handling (#3466) Glenn Jocher 2021-06-04 21:34:36 +02:00
• 563ea9475a Add check_git_status() 5 second timeout (#3464) Glenn Jocher 2021-06-04 21:17:36 +02:00
• c37f072ba7 Faster HSV augmentation (#3462) Yonghye Kwon 2021-06-05 03:02:20 +09:00
• d40481acc5 Add timeout() class (#3460) Glenn Jocher 2021-06-04 15:23:33 +02:00
• 8e3b4a0bf3 Update MixUp augmentation alpha=beta=32.0 (#3455) Glenn Jocher 2021-06-04 12:47:53 +02:00
• 4aa2959101 Suppress jit trace warning + graph once (#3454) Glenn Jocher 2021-06-04 12:37:41 +02:00
• af2bc3a1c3 Add FLOPs title to table (#3453) Glenn Jocher 2021-06-04 11:46:40 +02:00
• f8651c388f Parse URL authentication (#3424) Glenn Jocher 2021-06-03 18:44:58 +02:00
• 3cb9ad4fc4 Update FLOPs description (#3422) chocosajandBuildTools 2021-06-03 18:31:51 +08:00
• fdbe527dc0 Revert "cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379)" (#3395) Glenn Jocher 2021-05-31 10:39:00 +02:00
• d833ab3d25 Create is_pip() function (#3391) Glenn Jocher 2021-05-30 20:52:42 +02:00
• 4b52e19a61 COCO evolution fix (#3388) Glenn Jocher 2021-05-29 22:49:34 +02:00
• 21a9607e00 cv2.imread(img, -1) for IMREAD_UNCHANGED (#3379) tudouleiandGlenn Jocher 2021-05-30 03:12:01 +08:00
• 57f773b0ae Update tutorial.ipynb (#3368) Peretz Cohen 2021-05-29 11:49:24 -07:00
• ba6f3f974b Enable direct --weights URL definition (#3373) Glenn Jocher 2021-05-28 15:18:44 +02:00
• b78e30dd2b Merge remote-tracking branch 'origin/develop' into develop SkalskiP 2021-05-28 14:44:45 +02:00
• 3fea068384 update ci-testing.yml (#3322) Piotr Skalski 2021-05-27 17:01:36 +02:00
• bb131238aa update ci-testing.yml (#3322) Piotr Skalski 2021-05-27 17:01:36 +02:00
• 4d4a2b0520 Ignore blank lines in *.txt labels (#3366) Glenn Jocher 2021-05-27 14:31:26 +02:00
• ef4d53818d ONNX export in .train() mode fix (#3362) WangChaofeng 2021-05-27 20:10:14 +08:00