欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Pytorch

程序员文章站 2022-07-13 11:43:36
...

1. Python format 格式化函数:

 #Add the convolutional layer
 conv = nn.Conv2d(prev_filters, filters, kernel_size, stride, pad, bias = bias)
 module.add_module("conv_{0}".format(index), conv)
 #Add the Batch Norm Layer
 if batch_normalize:
     bn = nn.BatchNorm2d(filters)
     module.add_module("batch_norm_{0}".format(index), bn)

here using . formatto automatically increase the name of layers.

. format, 新增了一种格式化字符串的函数 str.format() the output is like

  (0): Sequential(
    (conv_0): Conv2d(3, 32, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
    (batch_norm_0): BatchNorm2d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (leaky_0): LeakyReLU(negative_slope=0.1, inplace

2. module.add_module

3. torch.cat(seq, dim=0, out=None) → Tensor

Concatenates the given sequence of seq tensors in the given dimension.

4. How to read images sequencetially

https://*.com/questions/6773584/how-is-pythons-glob-glob-ordered

import os, glob
def sortKeyFunc(s):           #used to sort the list by order
    return int(os.path.basename(s)[:-4])

myList = ["c:\tmp\x\123.csv", "c:\tmp\x\44.csv", "c:\tmp\x\101.csv", "c:\tmp\x\102.csv", "c:\tmp\x\12.csv"]
myList.sort(key=sortKeyFunc)

output as: ["c:\tmp\x\12.csv", "c:\tmp\x\44.csv", "c:\tmp\x\101.csv", "c:\tmp\x\102.csv", "c:\tmp\x\123.csv"]

5. How to rename files

https://github.com/sspatil89/renameFile/blob/master/renameFile.py

import os
"""
Renames the multiple file within the same directory with appending number
"""
path = 'Enter the directory path'
files = os.listdir(path)
i = 1

for file in files:
    filename, file_extension = os.path.splitext(file)
    os.rename(os.path.join(path, file), os.path.join(path, filename + str(i) + file_extension))
    i = i+1

6. Video2frames

args = arg_parse()
#### convert video to frames
#videofile = args.video
# cap = cv2.VideoCapture(videofile)
# assert cap.isOpened(), 'Cannot capture source'
#
# success, image = cap.read()
# count = 0
# success = True
#
# if not os.path.exists(args.det):
#     os.makedirs(args.det)
# directory = args.det
#
# print("Starting to convert ")
# while success:
#     success, image = cap.read()
#     cv2.imwrite(os.path.join(directory, "%d.jpg" % count), image)  # save frame as JPEG file
#     # cv2.imwrite("directory:\\frame%d.jpg" % count, image)     # save frame as JPEG file
#     if cv2.waitKey(10) == 27:  # exit if Escape is hit
#         break
# count += 1
# print("Completed video2frames")

7. read a dictionary and sort by the key

a = np.load('/home/biyi/PycharmProjects/SIMD/pytorch-yolo-v3/dic.npy').item()
od = sorted(a.items())
# print(od)