pytorch记录

跟"我是土堆"视频进行练习

"""
user='nike'
pwd='123'

user_input=input("your name:")
pwd_input=input("your pwd:")

if(user==user_input and pwd==pwd_input):
    print("login")
else:
    print("false")
"""


"""
import time

print('loadding',end=' ')
for i in range(6):
    print(".",end=' ')
    time.sleep(1)
"""

"""
age=10
while True:
    age_input=input(">>: ").strip()  #输入字符前后的空格进行删减
    if age_input.isdigit():  #判断字符串是否是数字型字符串
        age_input=int(age_input)   #转换成int
        if(age_input==age):
            print("you got it")
            break  #退出循环
"""

"""
#异常处理
s1='hello'
try:
    int(s1)
except IndexError as e:
    print(e)
except KeyError as e:
    print(e)
except ValueError as e:
    print(e)
"""

"""
age=19
count=0 #游戏次数控制
prize_dict={0:'布娃娃',1:'变形金刚',2:'奥特曼',3:'iiii'}

while(count<3):
    inp_age=input('请输入年龄>>>:') #与用户交互

    if not inp_age.isdigit():
        print("输入的不是数字")
        continue
    inp_age=int(inp_age)

    if(inp_age==age):
        print('猜对了')
        print(prize_dict)

        for i in range(2):
            prize_choice=input('请输入你想要的礼物,不想要,输入n退出:')
            if(prize_choice!='n'):
                print(f'恭喜获得奖品:{prize_dict[int(prize_choice)]}')
            else:
                break
        break
    elif inp_age<age:
        print('xiao')
    else:
        print('da')
    count+=1
    if(count!=3):
        continue

    again_choice=input('是否继续游戏,继续输入Y,否则按任意键退出.')
    if(again_choice=='Y'):
        count=0

"""

#tensorBoard的使用
from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image

writer = SummaryWriter("logs")  #把结果写入log文件夹中
Image_path="hymenoptera_data/hymenoptera_data/train/bees_image/17209602_fe5a5a746f.jpg"
img_PIL=Image.open(Image_path)  #根目录地址打开,img_PIL是<class 'PIL.JpegImagePlugin.JpegImageFile'>类型的
img_array=np.array(img_PIL)  #img_array为numpy形式
#出错了,则先看一下其类型
# print(type(img_array))  #输出:<class 'numpy.ndarray'> 没有错误,可以按ctrl+函数名字进行查询使用说明!!!
#看一下其形状
# print(img_array.shape) #输出:(512, 768, 3) 属于(H,W,3),通道在最后面
#从PIL到numpy,需要在add_image()中指定shape中每个数字/维表示的含义,即dataformats='HWC'
#tag (str)：数据标识符
#img_tensor（类型:torch.Tensor、numpy.ndarray 或 string/blobname）：图像数据
# global_step (int)：要记录的全局步值
#dataformats (str)：表格的图像数据格式规范 CHW、HWC、HW、WH 等
writer.add_image("test",img_array,2,dataformats='HWC')

# y=2x
for i in range(100):
    writer.add_scalar("y=2x",2*i,i)  #添加标量数据    tag:图的说明,参数2:x轴,参数3:y轴

writer.close()  #关闭