F:\室内跌倒数据\Annotations路径下为full_dataset_9209.xml文件,其中9209是序号;F:\室内跌倒数据\JPEGImages路径下为full_dataset_9209.jpg文件,其中9209为序号,请用python代码将F:\室内跌倒数据\Annotations路径下的full_dataset_9209与JPEGImages路径下相同的full_dataset_9209.jpg文件名提取出来放在F:\室内跌倒数据\img中
时间: 2023-11-30 11:05:06 浏览: 105
172张精挑细选的摔倒数据集,手工精细标注未摔倒和摔倒的两类,里面包含jpg图片和xml文件
5星 · 资源好评率100%
好的,您可以使用以下代码实现:
```python
import os
import shutil
# 设置路径
anno_path = r'F:\室内跌倒数据\Annotations'
img_path = r'F:\室内跌倒数据\JPEGImages'
output_path = r'F:\室内跌倒数据\img'
# 获取Annotations文件夹下的所有文件名
anno_files = os.listdir(anno_path)
# 遍历Annotations文件夹下的所有文件
for anno_file in anno_files:
# 获取文件名中的序号
index = anno_file.split('_')[-1].split('.')[0]
# 构造对应的JPEGImages文件名
img_file = 'full_dataset_{}.jpg'.format(index)
img_file_path = os.path.join(img_path, img_file)
# 判断JPEGImages文件是否存在,如果存在则将该文件复制到output_path目录下
if os.path.exists(img_file_path):
output_file_path = os.path.join(output_path, img_file)
shutil.copy(img_file_path, output_file_path)
```
这段代码会遍历Annotations文件夹下的所有文件,提取文件名中的序号,并根据序号构造对应的JPEGImages文件名。然后判断JPEGImages文件是否存在,如果存在则将该文件复制到output_path目录下。请确保代码中的路径正确,并根据实际情况进行修改。
阅读全文