Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity
682 Commits 18 Branches 0 Tags
ca290dca2427fef69b687d98ac99497c27cad144
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Glenn Jocher c8c5ef36c9 PyTorch 1.7.0 Compatibility Updates (#1233) 
* torch 1.7.0 compatibility updates

* add inference verification
 2020-10-28 15:03:50 +01:00
Glenn Jocher 453acdec67 Update tensorboard logging 2020-10-26 01:08:33 +01:00
Glenn Jocher 7f1640695b Update hubconf.py (#1210) 2020-10-26 01:06:54 +01:00
Glenn Jocher d3dad42256 Move generated *.json annotations into runs/test (#1209) 
* Move generated *.json annotations into runs/test

* list and string robust
 2020-10-25 18:19:44 +01:00
Glenn Jocher 0a52ae16dc README Update (#1207) 
* README Update

* Update README.md

* README Update

* Update README.md
 2020-10-25 14:05:28 +01:00
Glenn Jocher 8e9b3d0205 Precision-Recall Curve feature update #1206 2020-10-25 12:59:16 +01:00
Glenn Jocher ed85038c38 Precision-Recall Curve feature update (#1206) 
* Precision-Recall Curve feature update

* sentinel value update
 2020-10-25 12:55:52 +01:00
Glenn Jocher 7fe4a6bc06 FROM nvcr.io/nvidia/pytorch:20.10-py3 2020-10-25 10:49:33 +01:00
Glenn Jocher 481d46cffb Improved corruption handling during scan and cache (#999) 2020-10-24 15:09:19 +02:00
Glenn Jocher d61930e017 Improved corruption handling during scan and cache (#999) 2020-10-24 14:50:50 +02:00
Glenn Jocher 0fda95aaf4 Update cache out-of-date warning 2020-10-23 15:13:42 +02:00
Glenn Jocher 3b57cb5641 Simplified inference (#1153) 2020-10-15 20:10:08 +02:00
Glenn Jocher 4d3680c81d Minor import and spelling updates (#1133) 2020-10-13 17:24:27 +02:00
Glenn Jocher 4346b13a40 Dataset download bash script updates (#1132) 2020-10-13 15:58:03 +02:00
Glenn Jocher 0ada058f63 Generalized regression criterion renaming (#1120) 2020-10-11 17:25:17 +02:00
Glenn Jocher 10c85bf4eb FROM nvcr.io/nvidia/pytorch:20.09-py3 2020-10-11 16:25:14 +02:00
Glenn Jocher d45e349a1e Rearrange export input after checks (#1118) 
img size checks are warnings rather than errors, so current implementation allows improperly formed model inputs.
 2020-10-11 16:23:36 +02:00
Glenn Jocher d87cf7ecb9 Update requirements.txt coremltools==4.0 2020-10-11 16:21:39 +02:00
Glenn Jocher c2403d785c fuse() bug fix 2020-10-11 00:42:54 +02:00
Glenn Jocher 5fac5ad165 Precision-Recall Curve Feature Addition (#1107) 
* initial commit

* Update general.py

Indent update

* Update general.py

refactor duplicate code

* 200 dpi
 2020-10-09 14:50:59 +02:00
Glenn Jocher 9eae82e3a3 Update stale.yml exempt-issue-labels bug fix (#1098) 2020-10-08 16:41:04 +02:00
Glenn Jocher 5b8cf247af venv* ignore update 2020-10-06 15:17:12 +02:00
Glenn Jocher 124f0e8212 torchvision nms bug fix 2020-10-06 15:09:24 +02:00
Glenn Jocher 66676eb039 init_torch_seeds >> init_seeds bug fix 2020-10-06 15:00:47 +02:00
Glenn Jocher 883924d9dc classifier, export, torch seed updates 2020-10-06 14:54:02 +02:00
Glenn Jocher 35fe98543c Update requirements.txt coremltools==4.0b4 2020-10-05 01:47:04 +02:00
Glenn Jocher a12698f4c9 Update ci-testing.yml reflecting sys.path additions (#869) 
* Update ci-testing.yml

* Update yolo.py

* Update ci-testing.yml

* Update export.py

* Update yolo.py
 2020-10-04 18:50:32 +02:00
Glenn Jocher 4fce0096fa model.add_nms() method 2020-09-20 16:31:40 -07:00
Glenn Jocher 2f77cf33f6 .fuse() additional error checking 2020-09-20 12:04:20 -07:00
Glenn Jocher 89655a84f2 .fuse() gradient introduction bug fix 2020-09-20 11:57:19 -07:00
Glenn Jocher c4cb78570c add NMS to pretrained pytorch hub models 2020-09-18 18:17:11 -07:00
Glenn Jocher 5a9c5c1d3b add mosaic and warmup to hyperparameters (#931) 2020-09-13 14:08:06 -07:00
Glenn Jocher f1c63e2784 add mosaic and warmup to hyperparameters (#931) 2020-09-13 14:03:54 -07:00
Glenn Jocher 806e75f2b1 self.label_files bug fix (#958) 2020-09-13 11:38:23 -07:00
Glenn Jocher 6062bb7dee print self.yaml['nc'] override update 2020-09-11 17:44:11 -07:00
Glenn Jocher a62a45b2dd prevent testloader caching on --notest 2020-09-11 16:59:13 -07:00
Glenn Jocher c54e39464c check_file() assert unique 2020-09-11 14:33:06 -07:00
Glenn Jocher 720645d96a LoadImagesAndLabels() /images/ to /labels/ fix (#928) 2020-09-10 12:29:05 -07:00
Glenn Jocher d49c52eee3 _RepeatSampler outside of InfiniteDataLoader 2020-09-10 12:27:35 -07:00
Glenn Jocher bb8872ea5f Dockerfile sudo docker "kill all" bug fix 2020-09-08 00:56:35 -07:00
Glenn Jocher cab36f72a8 add .git to .dockerignore 2020-09-06 22:53:12 -07:00
Glenn Jocher 9c0efdde9f glob.glob(recursive=True) search (#917) 2020-09-06 22:52:28 -07:00
Glenn Jocher c8e51812a5 hyp evolution force-autoanchor fix 2020-09-04 13:13:10 -07:00
Glenn Jocher c687d5c129 reorganize train initialization steps 2020-09-04 12:25:53 -07:00
Glenn Jocher bc1fd13a7a gsutil cp hyp evolution bug fix (#876) 2020-09-03 20:29:08 -07:00
Glenn Jocher 44cdcc7e0b hyp['anchors'] evolution update 2020-09-03 12:54:22 -07:00
Glenn Jocher ffe9eb4238 remove if platform.system() == 'Darwin' checks 2020-09-03 11:41:33 -07:00
Glenn Jocher 23a7f29e2d Update hyp.finetune.yaml on 306 VOC generations result 2020-09-03 01:20:30 -07:00
Glenn Jocher 0a0675da12 Merge remote-tracking branch 'origin/master' 2020-09-03 01:07:48 -07:00
Glenn Jocher 354a97e4ef Create sotabench.py 2020-09-03 01:06:59 -07:00
Glenn Jocher 281d78c105 Update train.py (#902) 
* Update train.py with simplified ckpt names

* Return default hyps to hyp.scratch.yaml

Leave line commented for future use once mystery of best finetuning hyps to apply becomes clearer.

* Force test_batch*_pred.jpg replot on final epoch

This will allow you to see predictions final testing run after training completes in runs/exp0
 2020-09-02 15:08:43 -07:00
Glenn Jocher d68179953d Update export.py 2020-09-02 13:23:29 -07:00
Glenn Jocher d3f9bf2bb7 Update datasets.py 2020-09-01 17:02:47 -07:00
Glenn Jocher 3e04d20c7d Merge remote-tracking branch 'origin/master' 2020-08-31 11:07:00 -07:00
Glenn Jocher f06e2d518c opt.image_weights bug fix (#885) 2020-08-31 11:05:37 -07:00
Glenn Jocher 69ff781ca5 opt.img_weights bug fix (#885) 2020-08-31 10:33:07 -07:00
Glenn Jocher 987c226849 seaborn plotting correlogram addition 2020-08-29 15:13:58 -07:00
Glenn Jocher 08e97a2f88 Update hyperparameters to add lrf, anchors 2020-08-28 14:58:43 -07:00
Glenn Jocher 0c01afc390 Update Dockerfile to upgrade pip 2020-08-26 14:44:43 -07:00
Glenn Jocher 62e037419f Update Dockerfile FROM nvcr.io/nvidia/pytorch:20.08-py3 2020-08-26 00:44:34 -07:00
Glenn Jocher 31d6c389f2 Update general.py plot_evolution() 2020-08-25 23:08:14 -07:00
Glenn Jocher 1e7674d4dd Update export.py --img-size comment 2020-08-25 23:07:22 -07:00
Glenn Jocher a21bd0687c Update train.py forward simplification 2020-08-25 13:48:03 -07:00
Glenn Jocher 5a7d79fbe6 Update general.py bbox_iou() with eps improvements (#736) 
eps reduced to 1e-9 for additional precaution.
 2020-08-25 12:56:39 -07:00
Glenn Jocher 82ed33a052 generalize webcam detection with opt.source.isnumeric() (yolov3 #1462) 2020-08-25 11:07:38 -07:00
Glenn Jocher 09402a2174 torch.from_tensor() bug fix 2020-08-25 03:14:17 -07:00
Glenn Jocher 4fb8cb353f Update export.py with v3.0 Hardswish() support (#831) 2020-08-24 21:59:26 -07:00
Glenn Jocher 4d7f222f73 Update export.py with v3.0 Hardswish() support 2020-08-24 21:47:49 -07:00
Glenn Jocher 8666bc507e Create evolve.sh for infinite hyp evolution 2020-08-24 20:27:28 -07:00
Glenn Jocher a8751e50de model.fuse() fix for export.py (#827) 2020-08-24 19:27:54 -07:00
Glenn Jocher 5f07782f56 Update requirements.txt for coremltools kmeans quantization 2020-08-22 20:50:53 -07:00
Glenn Jocher 71209a6099 exportable Hardswish() implementation 2020-08-22 16:59:31 -07:00
Glenn Jocher fd71fe8451 Update hyp.finetune.yaml 2020-08-22 16:18:53 -07:00
Glenn Jocher f7e075fe93 gsutil cp yaml when evolving to GCP bucket 2020-08-22 15:47:39 -07:00
Glenn Jocher 9dbaa1bbbe Update plot_results() 2020-08-22 15:32:05 -07:00
Glenn Jocher 83dc540b1d remove ema.ema hasattr(ema, 'module') check 2020-08-22 15:18:39 -07:00
Glenn Jocher 1309400e6b Update hyp.finetune.yaml on VOC results 2020-08-22 15:05:47 -07:00
Glenn Jocher fb05681a04 Update requirements.txt to coremltools==4.0b3 2020-08-22 14:22:15 -07:00
Glenn Jocher d0f98c0095 Update yolo.py with model_info(verbose=False) 2020-08-21 16:20:45 -07:00
Glenn Jocher b9851222bf Merge remote-tracking branch 'origin/master' 2020-08-21 15:03:09 -07:00
Glenn Jocher c171e45cda Update test.py with set() (#806) 2020-08-21 15:03:03 -07:00
Glenn Jocher ea5e97fb9b FP16 NMS (#814) 2020-08-21 14:44:25 -07:00
Glenn Jocher b42e8a531b optimize imports 2020-08-20 21:17:40 -07:00
Glenn Jocher 4447f4b937 --resume to same runs/exp directory (#765) 
* initial commit

* add weight backup dir on resume
 2020-08-20 18:24:33 -07:00
Glenn Jocher 5e0b90de8f CIoU nan bug fix (#736) 2020-08-20 10:37:55 -07:00
Glenn Jocher da35727c68 --save-txt bug fix (#763) 2020-08-17 17:14:07 -07:00
Glenn Jocher ebafd1ead5 single command --resume (#756) 
* single command --resume

* else check files, remove TODO

* argparse.Namespace()

* tensorboard lr

* bug fix in get_latest_run()
 2020-08-17 16:28:43 -07:00
Glenn Jocher 2077d78880 update Dockerfile with libgl1-mesa-glx 2020-08-15 09:53:22 -07:00
Glenn Jocher f5da528d28 reformat code 2020-08-14 11:53:44 -07:00
Glenn Jocher b35f76e3d0 update .dockerignore 2020-08-14 11:29:15 -07:00
Glenn Jocher fe80bed695 Update stale.yml 2020-08-14 10:58:33 -07:00
Glenn Jocher 7b2b52193d update requirements.txt 2020-08-14 10:14:48 -07:00
Glenn Jocher 5991d144c1 Update README.md 2020-08-14 00:04:12 -07:00
Glenn Jocher 06bdd3955c Update README.md 2020-08-13 14:31:08 -07:00
Glenn Jocher 916d4aad9a v3.0 Release (#725) 
* initial commit

* remove yolov3-spp from test.py study

* update study --img range

* update mAP

* cleanup and speed updates

* update README plot
 2020-08-13 14:25:05 -07:00
Glenn Jocher 7875f4c1fb reverse plotting low to high confidence 2020-08-12 13:50:16 -07:00
Glenn Jocher d2da523053 Merge remote-tracking branch 'origin/master' 2020-08-11 20:11:58 -07:00
Glenn Jocher 9f482cbcb8 utils.general comment updates/bug fixes 2020-08-11 20:11:44 -07:00
Glenn Jocher e71fd0ec0b Model freeze capability (#679) 2020-08-10 22:49:43 -07:00
Glenn Jocher 8e5c66579b update train.py remove save_json final_epoch 2020-08-09 21:24:40 -07:00