Skip to content

Commit a2fdb33

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 90a1c82 commit a2fdb33

File tree

2 files changed

+5
-10
lines changed

2 files changed

+5
-10
lines changed

physical-ai-research/notebooks/02_monai_generation_test.ipynb

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -176,19 +176,19 @@
176176
"def sample_image(model, scheduler, num_inference_steps=50):\n",
177177
" \"\"\"Diffusion sampling (DDPM)\"\"\"\n",
178178
" model.eval()\n",
179-
" \n",
179+
"\n",
180180
" # Start from random noise\n",
181181
" image = torch.randn(1, 3, 256, 256).to(device)\n",
182-
" \n",
182+
"\n",
183183
" scheduler.set_timesteps(num_inference_steps)\n",
184-
" \n",
184+
"\n",
185185
" for t in scheduler.timesteps:\n",
186186
" # Predict noise\n",
187187
" noise_pred = model(image, timesteps=torch.tensor([t]).to(device))\n",
188-
" \n",
188+
"\n",
189189
" # Denoise step\n",
190190
" image = scheduler.step(noise_pred, t, image).prev_sample\n",
191-
" \n",
191+
"\n",
192192
" return image\n",
193193
"\n",
194194
"print(\"샘플링 시작 (학습되지 않은 모델이므로 노이즈만 생성됨)...\")\n",

physical-ai-research/scripts/download_datasets.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,8 @@
77
3. EndoVis - MICCAI Challenge Datasets
88
"""
99

10-
import os
1110
import argparse
12-
import urllib.request
1311
from pathlib import Path
14-
from typing import List
15-
import zipfile
16-
import gdown
1712

1813

1914
class DatasetDownloader:

0 commit comments

Comments
 (0)