Enable README.md + Update README.md (FIX) [10/17/2025]
#1
by
qpqpqpqpqpqp
- opened
- readme.md → README.md +8 -8
readme.md → README.md
RENAMED
|
@@ -4,19 +4,19 @@ Argument list:
|
|
| 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
|
| 12 |
```
|
| 13 |
python inference.py "model_path" "img1" "img2" [--save_path SAVE_PATH] [--gpu] [--fp16] [--frames FRAMES] [--fps FPS]
|
| 14 |
```
|
| 15 |
-
* ```model_path```
|
| 16 |
-
* ```img1```
|
| 17 |
-
* ```img2```
|
| 18 |
-
* ```--save_path SAVE_PATH```
|
| 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```
|
| 22 |
-
* ```--fps FPS``` FPS
|
|
|
|
| 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 the model immediately instead
|
| 8 |
* ```--gpu``` Whether to attempt to use GPU for testing
|
| 9 |
|
| 10 |
# Testing exported model
|
| 11 |
+
The following script creates a 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``` The path to the exported TorchScript checkpoint
|
| 16 |
+
* ```img1``` The path to the first image
|
| 17 |
+
* ```img2``` The path to the second image
|
| 18 |
+
* ```--save_path SAVE_PATH``` The 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``` The number of frames to interpolate between the input images
|
| 22 |
+
* ```--fps FPS``` FPS for the output video
|