Skip to content

Commit 9745024

Browse files
committed
update readme
1 parent 79b15f6 commit 9745024

File tree

1 file changed

+16
-19
lines changed

1 file changed

+16
-19
lines changed

README.md

Lines changed: 16 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -454,31 +454,25 @@ The following is a demo of image inference
454454
455455
456456
## 📝 Pixi Cheat Sheet
457-
Here are some useful tasks you can run with Pixi.
457+
Here are some useful tasks you can run with Pixi. You must install pixi on your machine first. See the [installation](#-installation) section for more details.
458+
For all the commands below, you can add `-e cuda` if you have a CUDA enabled GPU on your machine.
458459
459-
Run a quickstart
460+
Run a quickstart to check if your environment is setup correctly. By default, this will run in a CPU environment.
460461
```bash
461462
pixi run quickstart
462463
```
463464
464-
Smoke test the package
465-
```bash
466-
pixi run -e cpu quickstart
467-
```
465+
or to run in a CUDA enabled environment
468466
```bash
469467
pixi run -e cuda quickstart
470468
```
471469
472-
Train a model
473-
```bash
474-
pixi run -e cuda train-model
475-
```
476-
470+
Live inference with pretrained model on webcam
477471
```bash
478-
pixi run -e cpu train-model
472+
pixi run -e cuda live-inference-pretrained --webcam
479473
```
480474
481-
Run live inference
475+
Run live inference on a custom onnx model
482476
```bash
483477
pixi run -e cuda live-inference --onnx model.onnx --webcam --provider cuda --class-names classes.txt --inference-size 640
484478
```
@@ -495,11 +489,6 @@ pixi run -e cuda live-inference --onnx model.onnx --webcam --provider cuda --cla
495489
pixi run -e cpu live-inference --onnx model.onnx --input video.mp4 --class-names classes.txt --inference-size 320
496490
```
497491
498-
Live inference with pretrained model on webcam
499-
```bash
500-
pixi run -e cuda live-inference-pretrained --webcam
501-
```
502-
503492
Launch Gradio app
504493
```bash
505494
pixi run gradio-demo --model "best_prep.onnx" --classes "classes.txt" --examples "Rock Paper Scissors SXSW.v14i.coco/test"
@@ -509,12 +498,20 @@ pixi run gradio-demo --model "best_prep.onnx" --classes "classes.txt" --examples
509498
pixi run -e cpu gradio-demo
510499
```
511500
501+
Train a model
502+
```bash
503+
pixi run -e cuda train-model
504+
```
505+
506+
```bash
507+
pixi run -e cpu train-model
508+
```
509+
512510
Export model to ONNX
513511
```bash
514512
pixi run export --config config.yml --checkpoint model.pth --output model.onnx
515513
```
516514
517515
518-
519516
## ⚠️ Disclaimer
520517
I'm not affiliated with the original DEIM authors. I just found the model interesting and wanted to try it out. The changes made here are of my own. Please cite and star the original repo if you find this useful.

0 commit comments

Comments
 (0)