Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained(
5 "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
6)
7model = AutoModelForCausalLM.from_pretrained(
8 "taoki/deepseek-coder-6.7b-it-jmultiwoz-dolly-amenokaku-alpaca_jp_python"
9)
10
11if torch.cuda.is_available():
12 model = model.to("cuda")
13
14prompt="""あなたはプログラミングアシスタントです。
15### Instruction:
16OpenCVを用いて定点カメラから画像を保存するコードを示してください。
17### Response:
18"""
19
20input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
21outputs = model.generate(
22 **input_ids,
23 max_new_tokens=512,
24 do_sample=True,
25 top_p=0.9,
26 temperature=0.2,
27 repetition_penalty=1.1,
28)
29print(tokenizer.decode(outputs[0]))<|begin▁of▁sentence|>あなたはプログラミングアシスタントです。
### Instruction:
OpenCVを用いて定点カメラから画像を保存するコードを示してください。
### Response:
以下のPythonコードは、OpenCVを使用して定点カメラから画像を取得し、保存する方法を示します。
```python
import cv2
# カメラ設定
camera_id = 0 # 通常のカメラIDは0
cap = cv2.VideoCapture(camera_id)
if not cap.isOpened():
print("Failed to open camera")
else:
ret, frame = cap.read()
if ret:
# 画像を保存する
filename = 'image.jpg'
cv2.imwrite(filename, frame)
print('Image saved as', filename)
else:
print('Failed to capture image')
cap.release()
cv2.destroyAllWindows()
```
上記のコードでは、`cv2.VideoCapture()`関数でカメラを指定し、`cap.read()`で画像を読み込みます。成功した場合はTrueが返り、画像がframeとして返されます。
次に、`cv2.imwrite()`関数で画像を保存します。第一引数には保存先のパスを指定し、第二引数には保存する画像を指定します。
最後に、`cap.release()`でカメラを解放し、`cv2.destroyAllWindows()`で全てのウィンドウを破棄します。
<|EOT|>