1038lab commited on
Commit
3292ce8
·
verified ·
1 Parent(s): 61de08e

Update readme.me

Browse files
Files changed (1) hide show
  1. readme.me +14 -14
readme.me CHANGED
@@ -1,22 +1,22 @@
1
  Argument list:
2
 
3
- * ```model_path``` Path to the TF SavedModel
4
- * ```save_path``` Path to save the PyTorch state dict
5
- * ```--statedict``` Export to state dict instead of TorchScript
6
- * ```--fp32``` Save weights at full precision
7
- * ```--skiptest``` Skip testing and save model immediately instead
8
- * ```--gpu``` Whether to attempt to use GPU for testing
9
 
10
  # Testing exported model
11
  The following script creates an MP4 video of interpolated frames between 2 input images:
12
  ```
13
  python inference.py "model_path" "img1" "img2" [--save_path SAVE_PATH] [--gpu] [--fp16] [--frames FRAMES] [--fps FPS]
14
  ```
15
- * ```model_path``` Path to the exported TorchScript checkpoint
16
- * ```img1``` Path to the first image
17
- * ```img2``` Path to the second image
18
- * ```--save_path SAVE_PATH``` Path to save the interpolated frames as a video, if absent it will be saved in the same directory as ```img1``` is located and named ```output.mp4```
19
- * ```--gpu``` Whether to attempt to use GPU for predictions
20
- * ```--fp16``` Whether to use fp16 for calculations, speeds inference up on GPUs with tensor cores
21
- * ```--frames FRAMES``` Number of frames to interpolate between the input images
22
- * ```--fps FPS``` FPS of the output video
 
1
  Argument list:
2
 
3
+ * `model_path` Path to the TF SavedModel
4
+ * `save_path` Path to save the PyTorch state dict
5
+ * `--statedict` Export to state dict instead of TorchScript
6
+ * `--fp32` Save weights at full precision
7
+ * `--skiptest` Skip testing and save model immediately instead
8
+ * `--gpu` Whether to attempt to use GPU for testing
9
 
10
  # Testing exported model
11
  The following script creates an MP4 video of interpolated frames between 2 input images:
12
  ```
13
  python inference.py "model_path" "img1" "img2" [--save_path SAVE_PATH] [--gpu] [--fp16] [--frames FRAMES] [--fps FPS]
14
  ```
15
+ * `model_path` Path to the exported TorchScript checkpoint
16
+ * `img1` Path to the first image
17
+ * `img2` Path to the second image
18
+ * `--save_path SAVE_PATH` Path to save the interpolated frames as a video, if absent it will be saved in the same directory as ```img1``` is located and named ```output.mp4```
19
+ * `--gpu` Whether to attempt to use GPU for predictions
20
+ * `--fp16` Whether to use fp16 for calculations, speeds inference up on GPUs with tensor cores
21
+ * `--frames FRAMES` Number of frames to interpolate between the input images
22
+ * `--fps FPS` FPS of the output video