Commit Graph
100 Commits
Author SHA1 Message Date
Glenn Jocher 3004fb5bc1 Automatic m.half() profile on x.half() 2020-12-21 15:20:33 -08:00
Glenn Jocher 0bd9c48609 Update torch_utils.py
FLOPS to GFLOPS
2020-12-21 13:29:52 -08:00
Glenn Jocher 394d1c89f3 Input channel yaml['ch'] addition (#1741) 2020-12-19 10:54:01 -08:00
Glenn Jocher 685d601308 Increase plot_labels() speed (#1736) 2020-12-18 18:05:38 -08:00
Glenn Jocher 49abc722fc Update profile_idetection() (#1727) 2020-12-18 01:19:17 -08:00
Glenn Jocher d5289b54c4 clean_str() function addition (#1674)
* clean_str() function addition

* cleanup

* add euro symbol €

* add closing exclamation (spanish)

* cleanup
2020-12-17 17:20:20 -08:00
Glenn Jocher 7e161d9774 Single class train update (#1719) 2020-12-17 12:02:03 -08:00
Glenn Jocher 6bd5e8bca7 nn.SiLU() export support (#1713) 2020-12-16 17:55:57 -08:00
Glenn Jocher c923fbff90 W&B artifacts feature addition (#1712)
* Log artifacts

* cleanup
2020-12-16 17:52:12 -08:00
Glenn Jocher 8bc0027afc Update loss criteria constructor (#1711) 2020-12-16 08:39:35 -08:00
Glenn Jocher 799724108f Update C3 module (#1705) 2020-12-15 22:13:08 -08:00
Glenn Jocher 7947c86b57 Update COCO train postprocessing (#1702) 2020-12-15 21:50:28 -08:00
Glenn Jocher 69ea70cd3b Add idetection_profile() function to plots.py (#1700) 2020-12-15 18:35:47 -08:00
Glenn Jocher e92245a38c Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:42:27 -08:00
Glenn Jocher f7a923b446 Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:37:53 -08:00
Glenn Jocher 87ca35b922 Simplified PyTorch hub for custom models (#1677) 2020-12-12 11:16:57 -08:00
Glenn Jocher 54043a9fa4 Streaming --save-txt bug fix (#1672)
* Streaming --save-txt bug fix

* cleanup
2020-12-11 15:45:32 -08:00
Glenn Jocher bc52ea2d5f Profile() feature addition 2020-12-11 09:34:27 -08:00
Glenn Jocher ada90e3901 Profile() feature addition (#1673)
* Profile() feature addition

* cleanup
2020-12-11 09:30:39 -08:00
Glenn Jocher 94a7f55c4e FReLU bias=False bug fix (#1666) 2020-12-10 13:06:15 -08:00
Glenn Jocher b2bef8f6d8 vast.ai compatability updates (#1657) 2020-12-10 06:00:47 -08:00
Glenn Jocher 2e8e02745b vast.ai compatability updates (#1657) 2020-12-09 19:01:08 -08:00
Glenn Jocher fa8f1fb0e9 Simplify autoshape() post-process (#1653)
* Simplify autoshape() post-process

* cleanup

* cleanup
2020-12-09 07:44:06 -08:00
Glenn Jocher 84f9bb5d92 Normalized mosaic plotting bug fix (#1647) 2020-12-08 18:44:13 -08:00
Glenn Jocher 86f4247515 Hybrid auto-labelling support (#1646) 2020-12-08 18:15:39 -08:00
Glenn Jocher 0bb43953eb Reinstate PR curve sentinel values (#1645) 2020-12-08 17:40:49 -08:00
Glenn Jocher b53c8ac179 Create codeql-analysis.yml (#1644)
* Create codeql-analysis.yml

* Update ci-testing.yml
2020-12-08 17:03:16 -08:00
Glenn Jocher 68e6ab668b Hub device mismatch bug fix (#1619) 2020-12-06 17:53:38 +01:00
Glenn Jocher 791dadb51c Pycocotools best.pt after COCO train (#1616)
* Pycocotools best.pt after COCO train

* cleanup
2020-12-06 14:58:33 +01:00
Glenn Jocher d929bb656c Implement default class names (#1609) 2020-12-06 12:41:37 +01:00
Glenn Jocher efa7a915d8 Update download_weights.sh with usage example (#1613) 2020-12-06 10:09:09 +01:00
Glenn Jocher 8918e63476 Increase FLOPS robustness (#1608) 2020-12-05 11:41:34 +01:00
Glenn Jocher ba48f867ea Add bias to Classify() (#1601) 2020-12-04 15:06:33 +01:00
Glenn Jocher f010147578 Update matplotlib.use('Agg') tight (#1583)
* Update matplotlib tight_layout=True

* udpate

* udpate

* update

* png to ps

* update

* update
2020-12-02 15:53:16 +01:00
Glenn Jocher 784feae30a Update matplotlib svg backend (#1580) 2020-12-02 14:05:12 +01:00
Glenn Jocher b6ed1104a6 Daemon thread plotting (#1561)
* Daemon thread plotting

* remove process_batch

* plot after print
2020-11-30 16:44:14 +01:00
Glenn Jocher 68211f72c9 FROM nvcr.io/nvidia/pytorch:20.10-py3 (#1553)
* FROM pytorch/pytorch:latest

* FROM nvcr.io/nvidia/pytorch:20.10-py3
2020-11-29 17:47:51 +01:00
Glenn Jocher cff9263490 f.read().strip() (#1551) 2020-11-29 11:59:52 +01:00
Glenn Jocher 9fa7f9f598 f.read().strip() 2020-11-29 11:58:14 +01:00
Glenn Jocher 96a84468b9 Update labels_to_image_weights() (#1545) 2020-11-28 12:25:45 +01:00
Glenn Jocher 97a5227a1a Ignore W&B logging dir wandb/ (#1534) 2020-11-27 01:38:09 +01:00
Glenn Jocher 5dbf0c96f4 FROM nvcr.io/nvidia/pytorch:20.11-py3 2020-11-27 01:28:33 +01:00
Glenn Jocher c9798ae0e1 Update plot_study_txt() (#1533) 2020-11-26 22:18:17 +01:00
Glenn Jocher 0f2057ed33 Targets scaling bug fix (#1529) 2020-11-26 18:33:28 +01:00
Glenn Jocher 2c3efa430b Mosaic plots bug fix (#1526) 2020-11-26 14:02:22 +01:00
Glenn Jocher 12499f1c01 --image_weights bug fix (#1524) 2020-11-26 13:25:51 +01:00
Glenn Jocher 9728e2b8ae --image_weights bug fix (#1524) 2020-11-26 11:49:01 +01:00
Glenn Jocher e9a0ae6f19 Cache bug fix (#1513)
* Caching bug fix #1508

* np.zeros((0,5)) x2
2020-11-25 20:33:14 +01:00
Glenn Jocher 7d629fde05 Update README.md 2020-11-25 11:00:29 +01:00
Glenn Jocher d336b31f20 Update README.md 2020-11-25 11:00:00 +01:00
Glenn Jocher 2026d4c5eb Update caching (#1496) 2020-11-24 16:25:21 +01:00
Glenn Jocher bde5d9aaaa Update caching (#1496) 2020-11-24 16:23:00 +01:00
Glenn Jocher 0822cda781 Update caching (#1496) 2020-11-24 16:22:02 +01:00
Glenn Jocher 89c7a5b8dc Update caching (#1496) 2020-11-24 16:13:04 +01:00
Glenn Jocher 7aeef2dca5 Prevent PR plotting (#1489) 2020-11-24 00:46:01 +01:00
Glenn Jocher 0ddf12e108 Update tutorial.ipynb 2020-11-23 23:38:57 +01:00
Glenn Jocher bcac052eb9 Creado con Colaboratory 2020-11-23 23:37:23 +01:00
Glenn Jocher 8c0e7092a2 Update README.md 2020-11-23 19:28:35 +01:00
Glenn Jocher 354109c54c Autosplit (#1488) 2020-11-23 18:35:25 +01:00
Glenn Jocher 4798e66fdf Autosplit (#1488) 2020-11-23 17:18:21 +01:00
Glenn Jocher 0a3ff71ae0 Confusion matrix (#1474)
* initial commit

* add plotting

* matrix to cpu

* bug fix

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* seaborn pandas to requirements.txt

* seaborn pandas to requirements.txt

* update wandb plotting

* remove pandas

* if plots

* if plots

* if plots

* if plots

* if plots

* initial commit

* add plotting

* matrix to cpu

* bug fix

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* update plot

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* cleanup

* seaborn pandas to requirements.txt

* seaborn pandas to requirements.txt

* update wandb plotting

* remove pandas

* if plots

* if plots

* if plots

* if plots

* if plots

* Cat apriori to autolabels

* cleanup
2020-11-23 15:27:14 +01:00
Glenn Jocher 95fa65339f Cat apriori to autolabels (#1484) 2020-11-23 13:38:47 +01:00
Glenn Jocher 201bafc7cf Sync train and test iou_thresh (#1465)
* Sync train and test iou_thresh

* Sync train and test iou_thresh

* weights names .lower()

* Notebook update
2020-11-21 12:38:35 +01:00
Glenn Jocher d81bc47823 Creado con Colaboratory 2020-11-20 13:14:57 +01:00
Glenn Jocher a70e554fc3 Remove redundant downloads mirror (#1461) 2020-11-20 13:10:56 +01:00
Glenn Jocher 46c43b7b17 Creado con Colaboratory 2020-11-20 12:46:46 +01:00
Glenn Jocher 394131c2aa Use torchvision.ops.nms (#1460) 2020-11-20 11:23:36 +01:00
Glenn Jocher 199c9c7874 Detection() device bug fix (#1455) 2020-11-19 13:45:36 +01:00
Glenn Jocher af8aee76b3 Update greetings.yml (#1442)
* Update greetings.yml

* Update --feature-request.md

* Update --bug-report.md

* Rename --bug-report.md to bug-report.md

* Rename -question.md to question.md

* Rename --feature-request.md to feature-request.md

* Update greetings.yml

* Update greetings.yml
2020-11-19 00:26:33 +01:00
Glenn Jocher 225845e781 Update test.py for IoU in native image-space (#1439)
* Update test.py for IoU in native image-space

* remove redundant

* gn to device

* remove output scale_coords

* --img-size correction

* update

* native-space labels

* pred to predn

* remove clip_coords()
2020-11-18 22:50:21 +01:00
Glenn Jocher df0e408962 Merge remote-tracking branch 'origin/master' 2020-11-18 12:27:37 +01:00
Glenn Jocher 64bce3c822 Update labels.png with rectangles fix (#1432) 2020-11-18 12:27:30 +01:00
Glenn Jocher bd29a1c946 PosixPath to str webcam bug fix (#1437) 2020-11-18 10:03:41 +01:00
Glenn Jocher 1542ccadb2 Update labels.png with rectangles (#1432) 2020-11-18 00:17:33 +01:00
Glenn Jocher 412b113ac6 W&B logging bug fix (#1399) 2020-11-17 19:17:57 +01:00
Glenn Jocher 4250f84dfb Update PR curve (#1428)
* Update PR curve

* legend outside

* list(Path().glob())
2020-11-17 17:29:55 +01:00
Glenn Jocher 97c6a59dd1 PANet anchor order correction (#1420) 2020-11-17 10:13:00 +01:00
Glenn Jocher 311de00855 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:20:17 +01:00
Glenn Jocher 2aaaec3cca PyTorch Hub and autoShape update (#1415) 2020-11-16 23:16:35 +01:00
Glenn Jocher 7816c0c2d3 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:15:43 +01:00
Glenn Jocher 93fc607f07 PyTorch Hub and autoShape update (#1415) 2020-11-16 23:14:07 +01:00
Glenn Jocher f5429260ca PyTorch Hub and autoShape update (#1415)
* PyTorch Hub and autoShape update

* comment x for imgs

* reduce comment
2020-11-16 23:09:55 +01:00
Glenn Jocher 92c9b72832 Matplotlib AGG backend (#1413) 2020-11-16 16:34:07 +01:00
Glenn Jocher 548a98a391 Recursive directories dataset capability (#1408)
* Recursive directories dataset capability

* x.split('.')[-1]

* f += glob.glob(f"{p}{os.sep}**.*", recursive=True)

* f += glob.glob(str(p / '**.*'), recursive=True)

* remove . from image and vid formats

* .txt to txt

* str(p / '**' / '*.*')
2020-11-16 16:24:57 +01:00
Glenn Jocher 9c91aeae10 W&B logging add hyperparameters (#1399)
* W&B logging add hyperparameters

* hyp bug fix and image logging updates

* if plots and wandb:

* cleanup

* wandb/ gitignore add

* cleanup 2

* cleanup 3

* move wandb import to top of file

* wandb evolve

* update import

* wandb.run.finish()

* default anchors: 3
2020-11-16 13:35:34 +01:00
Glenn Jocher b7007d03b4 Creado con Colaboratory 2020-11-15 11:23:43 +01:00
Glenn Jocher 9f27902368 Improved FLOPS computation (#1398) 2020-11-14 14:48:55 +01:00
Glenn Jocher d8f5fcfe87 Improved FLOPS computation (#1398)
* Improved FLOPS computation

* update comment
2020-11-14 14:39:46 +01:00
Glenn Jocher 0c26c4e831 Increase pycocotools robustness (#1396) 2020-11-14 13:53:58 +01:00
Glenn Jocher 9b0f6e33ee Increase pycocotools robustness (#1396) 2020-11-14 13:09:04 +01:00
Glenn Jocher cf581305db Creado con Colaboratory 2020-11-14 12:07:31 +01:00
Glenn Jocher fe341fa44d Utils reorganization (#1392)
* Utils reorganization

* Add new utils files

* cleanup

* simplify

* reduce datasets.py

* remove evolve.sh

* loadWebcam cleanup
2020-11-14 11:50:32 +01:00
Glenn Jocher 379396e896 Yaml constructor posixpath --resume bug fix (#1390)
* resume fix for yaml constructor posixpath error

* fix update

* remove weights/ dir backup
2020-11-13 19:03:02 +01:00
Glenn Jocher c4addd7761 Unified '/project/name' results saving (#1377)
* Project/name update

* Update ci-testing.yml

* address project with path separator failure mode

* Project/name update

* address project with path separator failure mode

* Update ci-testing.yml

* detect.py default --name bug fix

* missing rstrip PR

* train/exp0 to train/exp
2020-11-12 23:37:46 +01:00
Glenn Jocher ce3872f86c W&B evolution bug fix (#1373) 2020-11-12 12:55:52 +01:00
Glenn Jocher f6b3c96966 Update tutorial.ipynb 2020-11-11 16:15:20 +01:00
Glenn Jocher 9cd46427e4 Creado con Colaboratory 2020-11-11 16:11:49 +01:00
Glenn Jocher 104f5418ea Creado con Colaboratory 2020-11-10 17:46:36 +01:00
Glenn Jocher 7841d7ba88 Update README.md 2020-11-10 17:12:35 +01:00
Glenn Jocher 6ff0d916fb Update download_weights.sh (#1338) 2020-11-10 15:04:14 +01:00