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

• 8efe97719c Add stop_training=False flag to callbacks (#6365) MatthiasandGlenn Jocher 2022-01-23 03:37:21 +01:00
• c43439aa31 Add albumentations to Dockerfile (#6392) Glenn Jocher 2022-01-21 21:06:02 -10:00
• bd815d48df FROM nvcr.io/nvidia/pytorch:21.10-py3 (#6379) Glenn Jocher 2022-01-20 19:15:04 -10:00
• 1b41a1d059 FROM nvcr.io/nvidia/pytorch:21.12-py3 (#6377) Glenn Jocher 2022-01-20 17:59:41 -10:00
• 9bcc32a5bf Flush callbacks when on --evolve (#6374) Ayush ChaurasiaandGlenn Jocher 2022-01-21 04:52:47 +05:30
• e1893c894a DEVICE_COUNT instead of WORLD_SIZE to calculate nw (#6324) sitecao 2022-01-20 18:06:26 -05:00
• 4e841b9b16 Reuse de_parallel() rather than is_parallel() (#6354) imyhxy 2022-01-21 04:50:17 +08:00
• 9708cf56ea New environment variable VERBOSE (#6353) johnk2hawaii 2022-01-19 14:32:19 -10:00
• 750c42e43e export.py automatic forward_export (#6352) Glenn Jocherandpre-commit-ci[bot] 2022-01-19 12:24:40 -10:00
• e5219099cd Created using Colaboratory Glenn Jocher 2022-01-19 10:18:29 -10:00
• 0cf932bf63 export.py return exported files/dirs (#6343) Glenn Jocher 2022-01-18 15:18:23 -10:00
• e2e95b2d8e TensorRT assert im.device.type != 'cpu' on export (#6340) Glenn Jocher 2022-01-18 13:52:25 -10:00
• fd55271c04 Console corrupted -> corrupt (#6338) Glenn Jocher 2022-01-18 10:49:26 -10:00
• 3119b2f27c Remove dataset_stats() autodownload capability (#6303) Glenn Jocher 2022-01-18 10:04:49 -10:00
• db1f83be63 Fix train.py parameter groups desc error (#6318) OtfotandGlenn Jocher 2022-01-17 14:58:34 +08:00
• 436ffc417a select_device() cleanup (#6302) Glenn Jocher 2022-01-14 15:48:15 -10:00
• a1a9c6884c Fixing bug multi-gpu training (#6299) Henry 2022-01-14 21:11:06 +01:00
• e7bf38277f Fix device count check (#6290) Glenn Jocher 2022-01-13 21:23:03 -10:00
• af001349e4 Add is_kaggle() function (#6285) Glenn Jocher 2022-01-13 12:39:42 -10:00
• 80473a6551 Update export.py with Detect, Validate usages (#6280) Glenn Jocher 2022-01-12 15:48:40 -10:00
• f3085accd3 Enable ONNX --half FP16 inference (#6268) Glenn Jocherandpre-commit-ci[bot] 2022-01-11 10:13:17 -10:00
• b3eaf5008b TensorRT pip install Glenn Jocher 2022-01-10 16:49:10 -10:00
• 9b13a594e9 Fix cmd string on tfjs export (#6243) Jinwoong YooandGlenn Jocher 2022-01-10 15:40:47 +09:00
• 6865d19a92 TensorRT 7 export fix (#6235) Glenn Jocher 2022-01-07 09:31:17 -08:00
• 33a67b4918 Update P2-P7 models/hub variants (#6230) Glenn Jocher 2022-01-06 11:08:09 -08:00
• ad565e31d2 Update README speed reproduction command (#6228) Glenn Jocher 2022-01-06 09:55:31 -08:00
• f80c463010 Attempt edgetpu-compiler autoinstall (#6223) Glenn Jocher 2022-01-05 20:57:20 -08:00
• b4ac3df6ff Add edgetpu_compiler checks (#6218) Glenn Jocher 2022-01-05 14:55:04 -08:00
• 00d7b97869 TFLite --int8 'flatbuffers==1.12' fix 2 (#6217) Glenn Jocher 2022-01-05 13:34:36 -08:00
• 8125ec5d42 TFLite --int8 'flatbuffers==1.12' fix (#6216) Glenn Jocher 2022-01-05 13:01:21 -08:00
• 5402753a53 Edge TPU compiler comment (#6196) Glenn Jocher 2022-01-04 19:36:12 -08:00
• 9e9219fe17 Fix nan-robust stream FPS (#6198) Glenn Jocher 2022-01-04 19:32:42 -08:00
• b5b56a3c88 Add CoreML inference (#6195) Glenn Jocher 2022-01-04 17:49:09 -08:00
• 7b31a531b4 Add tensorrt>=7.0.0 checks (#6193) Glenn Jocher 2022-01-04 13:39:13 -08:00
• a2f4a1799b TensorRT 7 anchor_grid compatibility fix (#6185) imyhxyandGlenn Jocher 2022-01-05 03:09:25 +08:00
• fb839298a1 Fix TorchScript on mobile export (#6183) 3 people 2022-01-04 12:25:48 +08:00
• 5bd6a97b18 Global export format sort (#6182) Glenn Jocher 2022-01-03 20:08:15 -08:00
• 7cad6597bb Ignore *_openvino_model/ dir (#6180) Glenn Jocher 2022-01-03 18:43:21 -08:00
• 63a4d862aa Add OpenVINO inference (#6179) Glenn Jocher 2022-01-03 15:41:26 -08:00
• b4a29b5a8d Update NMS max_wh=7680 for 8k images (#6178) Glenn Jocher 2022-01-03 10:54:52 -08:00
• 5344e54da6 [pre-commit.ci] pre-commit suggestions (#6177) pre-commit-ci[bot] 2022-01-03 10:42:50 -08:00
• 968e30065a Update greetings.yml (#6165) Glenn Jocher 2022-01-02 19:47:03 -08:00
• ec4b6dd2a3 Update export format docstrings (#6151) Glenn Jocherandpre-commit-ci[bot] 2022-01-02 16:09:45 -08:00
• e1dc894364 Enable AdamW optimizer (#6152) bilzard 2022-01-03 06:10:19 +09:00
• d95978a562 Add EdgeTPU support (#3630) 4 people 2022-01-01 01:47:52 +08:00
• affa284352 Refactor/reduce G/C/D/IoU if: else statements (#6087) 3 people 2021-12-31 05:59:29 +08:00
• 7b6938d5b5 Log best results (#6085) Awsaf 2021-12-31 03:47:53 +06:00
• 9155eb8641 W&B: Log best results after training ends (#6120) Ayush ChaurasiaandGlenn Jocher 2021-12-31 01:13:19 +05:30
• db6ec66a60 W&B: track batch size after autobatch (#6039) Ayush ChaurasiaandGlenn Jocher 2021-12-23 05:23:50 -08:00
• c72270c076 Init tensor directly on device (#6068) Deep Patel 2021-12-23 07:49:00 -05:00
• afa5cfb0f8 Reduce G/D/CIoU logic operations (#6074) JieLiand李杰 2021-12-23 18:53:00 +08:00
• 95c7bc25d3 OpenVINO Export (#6057) Glenn Jocher 2021-12-22 20:29:48 +01:00
• dc54ed5763 --freeze fix (#6044) Glenn Jocher 2021-12-20 18:24:07 +01:00
• b8a4babd60 Simplify set_logging() indexing (#6042) Glenn Jocher 2021-12-20 17:42:52 +01:00
• 0db9d5b6a2 Kaggle LOGGER fix (#6041) Glenn Jocher 2021-12-20 17:30:46 +01:00
• 26f0415287 Add dataset source citations (#6032) Glenn Jocher 2021-12-19 15:19:04 +01:00
• abbdd4802e train -> val comment fix (#6024) Glenn Jocher 2021-12-17 16:43:00 +01:00
• 361705d9be Multi-layer capable --freeze argument (#6019) Felix YouandGlenn Jocher 2021-12-17 22:42:26 +08:00
• 407a905747 Check TensorRT>=8.0.0 version (#6021) Glenn Jocherandpre-commit-ci[bot] 2021-12-17 14:59:46 +01:00
• c1249a47c7 *.torchscript inference self.jit fix (#6007) Glenn Jocher 2021-12-16 14:10:54 +01:00
• 628817dfae Recommend jar xf file.zip for zips (#5993) Glenn Jocher 2021-12-15 17:19:19 +01:00
• da9a1b719b Allow --weights URL (#5991) Glenn Jocher 2021-12-15 15:27:08 +01:00
• b7d18f3636 Increase ar_thr from 20 to 100 for better detection on slender (high aspect ratio) objects (#5556) Mrinal Jain 2021-12-15 19:12:23 +05:30
• c9a46a60e0 Update callbacks.py with __init__() (#5979) Glenn Jocher 2021-12-14 15:47:49 +01:00
• d699c21c75 YOLOv5s6 params and FLOPs fix (#5977) Glenn Jocher 2021-12-14 11:24:39 +01:00
• 2d0c6afbfe make parameter ignore epochs (#5972) jinmcandGlenn Jocher 2021-12-14 19:18:34 +09:00
• e8ef8fb1ca pretrained=False fix (#5966) Glenn Jocher 2021-12-13 13:32:27 +01:00
• 19c56e60b1 Fix imgsz bug (#5948) Diego MontesandGlenn Jocher 2021-12-12 17:39:14 -05:00
• 8f354362cd Fix Detections class tolist() method (#5945) 3 people 2021-12-11 18:40:37 +01:00
• 8f875d93a2 Refactor NUM_THREADS (#5954) Glenn Jocher 2021-12-11 14:32:18 +01:00
• 2c6317547a Add nms and agnostic nms to export.py (#5938) Diego MontesandGlenn Jocher 2021-12-10 12:24:32 -05:00
• a42af30d8a Update strip_optimizer() (#5949) iumyx2612 2021-12-11 00:06:27 +07:00
• 922fbd8209 fix .gitignore not tracking existing folders (#5946) Pascal MaillardandGlenn Jocher 2021-12-10 17:32:09 +01:00
• c45f9f678d Make select_device() robust to batch_size=-1 (#5940) Felix YouandGlenn Jocher 2021-12-10 22:27:20 +08:00
• 4fb6dd4b26 Fix ONNX opset inconsistency with parseargs and run args (#5937) Diego Montes 2021-12-09 17:10:16 -05:00
• 5bdb28ed10 Default PyTorch Hub to autocast(False) (#5926) Glenn Jocher 2021-12-08 23:15:14 +01:00
• c77a5a84e3 Absolute '/content/sample_data' (#5922) Glenn Jocher 2021-12-08 17:31:31 +01:00
• a3d5f1d3e3 Revert "Update plot_lr_scheduler() (#5864)" (#5920) Glenn Jocher 2021-12-08 16:46:24 +01:00
• 7d56d45124 Add hardware checks to notebook_init() (#5919) Glenn Jocherandpre-commit-ci[bot] 2021-12-08 14:57:03 +01:00
• 581dc301a7 Add ONNX inference providers (#5918) Glenn Jocher 2021-12-08 13:37:33 +01:00
• 554f782537 Add *.engine (TensorRT extensions) to .gitignore (#5911) greg2451andGlenn Jocher 2021-12-07 18:20:16 +01:00
• 3f152e5807 Update wandb_utils.py (#5908) Ayush Chaurasia 2021-12-07 20:39:11 +05:30
• fa05f8c977 Detections().tolist() explicit argument fix (#5907) Li Zeng 2021-12-07 16:01:41 +01:00
• 1075488d89 Single-command multiple-model export (#5882) Glenn Jocher 2021-12-04 16:28:40 +01:00
• 7bf04d9bbf AutoShape() models as DetectMultiBackend() instances (#5845) Glenn Jocher 2021-12-04 15:00:07 +01:00
• d885799c71 Update nl after cutout() (#5873) Glenn Jocher 2021-12-03 15:28:14 +01:00
• 360eec6910 Update plot_lr_scheduler() (#5864) Can 2021-12-03 20:37:45 +08:00
• 92a7391039 Add --workers 8 argument to val.py (#5857) iumyx2612andGlenn Jocher 2021-12-02 22:49:50 +07:00
• 00e308f7be Update TorchScript suffix to *.torchscript (#5856) Glenn Jocher 2021-12-02 16:06:45 +01:00
• 30db14fea8 Update val.py (#5838) Vishnu PradeepandGlenn Jocher 2021-12-02 13:20:58 +01:00
• 1679aacdc7 Update LoadImages ret_val=False handling (#5852) gggmt 2021-12-02 17:57:39 +08:00
• e8f8f2b903 Avoid inplace modifyingimgs in LoadStreams (#5850) Yu Zhang 2021-12-02 17:51:19 +08:00
• bc484579d7 Handle non-TTY wandb.errors.UsageError (#5839) Glenn Jocher 2021-12-01 15:38:02 +01:00
• a4207a202d Fix TensorRT potential unordered binding addresses (#5826) imyhxy 2021-11-30 20:52:22 +08:00
• 5ca5dd4c87 Update dataset_stats() to cv2.INTER_AREA (#5821) Glenn Jocher 2021-11-29 13:15:36 +01:00
• 8277033b65 Bump actions/cache from 2.1.6 to 2.1.7 (#5816) dependabot[bot] 2021-11-29 12:04:36 +01:00
• 94d8fec6d8 GCP VM from Image example (#5814) Glenn Jocher 2021-11-28 18:12:46 +01:00
• fcd180d336 Refactor new model.warmup() method (#5810) Glenn Jocher 2021-11-27 12:29:45 +01:00
• 7c6bae0ae6 Remove NCOLS from tqdm (#5804) Glenn Jocherandpre-commit-ci[bot] 2021-11-26 13:37:28 +01:00
• 53349dac8e Scope TF imports in DetectMultiBackend() (#5792) Phil2020andGlenn Jocher 2021-11-25 16:54:00 +00:00