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

• 7220cee1d1 mosaic4 bottom left image fix (#1021) haoyuefan 2020-09-24 00:51:33 +08:00
• 702c4fa53e adding the configuration to deploy yolov5 in in app engine (#964) Guilhenandguilhembau 2020-09-22 11:48:44 -07:00
• b8f656742d feat: add rtmp support (#1009) 3 people 2020-09-23 02:37:49 +08:00
• 4fce0096fa model.add_nms() method Glenn Jocher 2020-09-20 16:31:40 -07:00
• 2f77cf33f6 .fuse() additional error checking Glenn Jocher 2020-09-20 12:04:20 -07:00
• 89655a84f2 .fuse() gradient introduction bug fix Glenn Jocher 2020-09-20 11:57:19 -07:00
• c4cb78570c add NMS to pretrained pytorch hub models Glenn Jocher 2020-09-18 18:17:11 -07:00
• 5a9c5c1d3b add mosaic and warmup to hyperparameters (#931) Glenn Jocher 2020-09-13 14:08:06 -07:00
• f1c63e2784 add mosaic and warmup to hyperparameters (#931) Glenn Jocher 2020-09-13 14:03:54 -07:00
• 806e75f2b1 self.label_files bug fix (#958) Glenn Jocher 2020-09-13 11:38:23 -07:00
• 6062bb7dee print self.yaml['nc'] override update Glenn Jocher 2020-09-11 17:44:11 -07:00
• a62a45b2dd prevent testloader caching on --notest Glenn Jocher 2020-09-11 16:59:13 -07:00
• 35dc935504 remove nn.Conv2d in default settings (#949) tomguluson92 2020-09-12 07:35:48 +08:00
• c54e39464c check_file() assert unique Glenn Jocher 2020-09-11 14:33:06 -07:00
• 720645d96a LoadImagesAndLabels() /images/ to /labels/ fix (#928) Glenn Jocher 2020-09-10 12:29:05 -07:00
• d49c52eee3 _RepeatSampler outside of InfiniteDataLoader Glenn Jocher 2020-09-10 12:27:35 -07:00
• bb8872ea5f Dockerfile sudo docker "kill all" bug fix Glenn Jocher 2020-09-08 00:56:35 -07:00
• cab36f72a8 add .git to .dockerignore Glenn Jocher 2020-09-06 22:53:12 -07:00
• 9c0efdde9f glob.glob(recursive=True) search (#917) Glenn Jocher 2020-09-06 22:52:28 -07:00
• c8e51812a5 hyp evolution force-autoanchor fix Glenn Jocher 2020-09-04 13:13:10 -07:00
• c687d5c129 reorganize train initialization steps Glenn Jocher 2020-09-04 12:25:53 -07:00
• bc1fd13a7a gsutil cp hyp evolution bug fix (#876) Glenn Jocher 2020-09-03 20:29:08 -07:00
• 44cdcc7e0b hyp['anchors'] evolution update Glenn Jocher 2020-09-03 12:54:22 -07:00
• ffe9eb4238 remove if platform.system() == 'Darwin' checks Glenn Jocher 2020-09-03 11:41:33 -07:00
• 23a7f29e2d Update hyp.finetune.yaml on 306 VOC generations result Glenn Jocher 2020-09-03 01:20:30 -07:00
• 0a0675da12 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-09-03 01:07:48 -07:00
• 354a97e4ef Create sotabench.py Glenn Jocher 2020-09-03 01:06:59 -07:00
• d8274d0434 Fix results_file not renaming (#903) NanoCode012 2020-09-03 14:47:50 +07:00
• 281d78c105 Update train.py (#902) Glenn Jocher 2020-09-02 15:08:43 -07:00
• d68179953d Update export.py Glenn Jocher 2020-09-02 13:23:29 -07:00
• 6f3db5e662 Remove autoanchor and class checks on resumed training (#889) Naman GuptaandGlenn Jocher 2020-09-02 21:35:31 +05:30
• c5969f79f5 add support for DWConv fuse operation (#898) tkianai 2020-09-02 13:18:42 +08:00
• d3f9bf2bb7 Update datasets.py Glenn Jocher 2020-09-01 17:02:47 -07:00
• 901243c780 Fix relative path for VOC download (#880) NanoCode012 2020-09-01 06:06:10 +07:00
• 3e04d20c7d Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-08-31 11:07:00 -07:00
• f06e2d518c opt.image_weights bug fix (#885) Glenn Jocher 2020-08-31 11:05:37 -07:00
• 1e15aad6f9 Add InfiniteDataLoader class (#876) NanoCode012 2020-09-01 01:01:25 +07:00
• 69ff781ca5 opt.img_weights bug fix (#885) Glenn Jocher 2020-08-31 10:33:07 -07:00
• 987c226849 seaborn plotting correlogram addition Glenn Jocher 2020-08-29 15:13:58 -07:00
• 08e97a2f88 Update hyperparameters to add lrf, anchors Glenn Jocher 2020-08-28 14:58:43 -07:00
• 9776e70988 torch.ops.torchvision.nms (#860) rafale77 2020-08-27 11:13:41 -07:00
• 0c01afc390 Update Dockerfile to upgrade pip Glenn Jocher 2020-08-26 14:44:43 -07:00
• 62e037419f Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3 Glenn Jocher 2020-08-26 00:44:34 -07:00
• 31d6c389f2 Update general.py plot_evolution() Glenn Jocher 2020-08-25 23:08:14 -07:00
• 1e7674d4dd Update export.py --img-size comment Glenn Jocher 2020-08-25 23:07:22 -07:00
• a21bd0687c Update train.py forward simplification Glenn Jocher 2020-08-25 13:48:03 -07:00
• 455f7b8f76 Update general.py (#823) Willie Maddox 2020-08-25 15:01:17 -05:00
• 5a7d79fbe6 Update general.py bbox_iou() with eps improvements (#736) Glenn Jocher 2020-08-25 12:56:39 -07:00
• 82ed33a052 generalize webcam detection with opt.source.isnumeric() (yolov3 #1462) Glenn Jocher 2020-08-25 11:07:38 -07:00
• 09402a2174 torch.from_tensor() bug fix Glenn Jocher 2020-08-25 03:14:17 -07:00
• 4fb8cb353f Update export.py with v3.0 Hardswish() support (#831) Glenn Jocher 2020-08-24 21:59:26 -07:00
• 4d7f222f73 Update export.py with v3.0 Hardswish() support Glenn Jocher 2020-08-24 21:47:49 -07:00
• 8666bc507e Create evolve.sh for infinite hyp evolution Glenn Jocher 2020-08-24 20:27:28 -07:00
• a8751e50de model.fuse() fix for export.py (#827) Glenn Jocher 2020-08-24 19:27:54 -07:00
• 5f07782f56 Update requirements.txt for coremltools kmeans quantization Glenn Jocher 2020-08-22 20:50:53 -07:00
• 71209a6099 exportable Hardswish() implementation Glenn Jocher 2020-08-22 16:59:31 -07:00
• fd71fe8451 Update hyp.finetune.yaml Glenn Jocher 2020-08-22 16:18:53 -07:00
• f7e075fe93 gsutil cp yaml when evolving to GCP bucket Glenn Jocher 2020-08-22 15:47:39 -07:00
• 9dbaa1bbbe Update plot_results() Glenn Jocher 2020-08-22 15:32:05 -07:00
• 83dc540b1d remove ema.ema hasattr(ema, 'module') check Glenn Jocher 2020-08-22 15:18:39 -07:00
• 1309400e6b Update hyp.finetune.yaml on VOC results Glenn Jocher 2020-08-22 15:05:47 -07:00
• fb05681a04 Update requirements.txt to coremltools==4.0b3 Glenn Jocher 2020-08-22 14:22:15 -07:00
• d0f98c0095 Update yolo.py with model_info(verbose=False) Glenn Jocher 2020-08-21 16:20:45 -07:00
• b9851222bf Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-08-21 15:03:09 -07:00
• c171e45cda Update test.py with set() (#806) Glenn Jocher 2020-08-21 15:03:03 -07:00
• ea5e97fb9b FP16 NMS (#814) Glenn Jocher 2020-08-21 14:44:25 -07:00
• b42e8a531b optimize imports Glenn Jocher 2020-08-20 21:17:40 -07:00
• 4447f4b937 --resume to same runs/exp directory (#765) Glenn Jocher 2020-08-20 18:24:33 -07:00
• 5e0b90de8f CIoU nan bug fix (#736) Glenn Jocher 2020-08-20 10:37:55 -07:00
• fb4fc8cd02 Fix ema attribute error in DDP mode (#775) NanoCode012andGlenn Jocher 2020-08-18 23:52:21 +07:00
• da35727c68 --save-txt bug fix (#763) Glenn Jocher 2020-08-17 17:14:07 -07:00
• ebafd1ead5 single command --resume (#756) Glenn Jocher 2020-08-17 16:28:43 -07:00
• 26c3b11fdc Update torch_utils.py (#751) Francisco Ingham 2020-08-16 17:58:59 -03:00
• 2077d78880 update Dockerfile with libgl1-mesa-glx Glenn Jocher 2020-08-15 09:53:22 -07:00
• f5da528d28 reformat code Glenn Jocher 2020-08-14 11:53:44 -07:00
• b35f76e3d0 update .dockerignore Glenn Jocher 2020-08-14 11:29:15 -07:00
• fe80bed695 Update stale.yml Glenn Jocher 2020-08-14 10:58:33 -07:00
• 7b2b52193d update requirements.txt Glenn Jocher 2020-08-14 10:14:48 -07:00
• 5991d144c1 Update README.md Glenn Jocher 2020-08-14 00:04:12 -07:00
• 66744a0df1 Update utils/google_utils.py (#733) Quan Lin 2020-08-14 11:44:14 +10:00
• 06bdd3955c Update README.md Glenn Jocher 2020-08-13 14:31:08 -07:00
• 916d4aad9a v3.0 Release (#725) Glenn Jocher 2020-08-13 14:25:05 -07:00
• 1ddf692bf8 Update readme to remove apex dependency (#727) tkianai 2020-08-13 12:18:29 +08:00
• 56c2c344ff Fix list paths (#721) HatovixandGlenn Jocher 2020-08-13 03:53:38 +02:00
• 0892c44bc4 Fix Logging (#719) NanoCode012andGlenn Jocher 2020-08-13 04:18:19 +07:00
• a925f283a7 max workers for dataloader (#722) Marc 2020-08-12 16:57:36 -04:00
• 7875f4c1fb reverse plotting low to high confidence Glenn Jocher 2020-08-12 13:50:16 -07:00
• d2da523053 Merge remote-tracking branch 'origin/master' Glenn Jocher 2020-08-11 20:11:58 -07:00
• 9f482cbcb8 utils.general comment updates/bug fixes Glenn Jocher 2020-08-11 20:11:44 -07:00
• 1f92422e20 Github release assets model autodownload (#711) MarcandGlenn Jocher 2020-08-11 20:17:24 -04:00
• 4949401a94 Fix redundant outputs via Logging in DDP training (#500) NanoCode012andGlenn Jocher 2020-08-12 01:18:45 +07:00
• 542833c997 Filter the None labels (#705) tkianai 2020-08-12 02:15:48 +08:00
• e71fd0ec0b Model freeze capability (#679) Glenn Jocher 2020-08-10 22:49:43 -07:00
• 9ae868364a the value should be a list to unpacking (#688) tkianai 2020-08-11 02:23:48 +08:00
• 8e5c66579b update train.py remove save_json final_epoch Glenn Jocher 2020-08-09 21:24:40 -07:00
• 41523e2c91 Dataset autodownload feature addition (#685) Glenn Jocher 2020-08-09 20:52:57 -07:00
• 3d8ed0a76b Fix missing model.stride in DP and DDP mode (#683) NanoCode012 2020-08-10 01:01:36 +07:00
• a0ac5adb7b Single-source training update (#680) Glenn Jocher 2020-08-09 02:27:35 -07:00
• 3c6e2f7668 Single-source training (#680) Glenn Jocher 2020-08-09 02:12:44 -07:00
• d7cfbc47ab Fix unrecognized local rank argument (#676) NanoCode012 2020-08-09 06:40:10 +07:00