更改--img-size,需要更改yolov7代码的哪些地方
时间: 2023-11-15 18:06:04 浏览: 84
更改 `--img-size` 参数后,需要在以下几个地方更改 YOLOv7 代码:
1. `models/yolo.py`:更改 `__init__` 方法中的 `self.img_size` 变量,将其设置为新的图片尺寸。如下所示:
```python
class YOLOv7(nn.Module):
def __init__(self, img_size=640, # 更改这里的img_size变量
...
):
super(YOLOv7, self).__init__()
self.img_size = img_size # 更改这里的img_size变量
...
```
2. `utils/datasets.py`:更改 `load_image` 方法中的 `resize` 方法,将其设置为新的图片尺寸。如下所示:
```python
def load_image(self, index):
...
# 更改下面这行代码中的resize参数为新的图片尺寸
img = img.resize((self.img_size, self.img_size), resample=Image.BILINEAR)
...
```
3. `detect.py`:更改 `detect` 方法中的 `img_size` 变量,将其设置为新的图片尺寸。如下所示:
```python
def detect(save_img=False):
...
# 更改下面这行代码中的img_size参数为新的图片尺寸
img_size = opt.img_size
...
```
这些是需要更改的主要代码部分,如果还有其他代码部分依赖于图片尺寸,请根据需要进行更改。
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![-](https://csdnimg.cn/download_wenku/file_type_column_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)