Closed MiniBullLab closed 3 years ago
classnet中,换backbone: 由resnet18换成mobilenetv2_120d网络,在caltech257_classify数据集中,精度64.1提升到85.0。
anchors的大小目前在object365上重新做了聚类,anchors为10,10|24,26|33,68|61,131|70,36|117,78|128,209|277,107|311,251。
目前常用的数据增强方式:
我们工程中的数据增强方式:
classnet任务中的数据增强包括:
拉取最新分支代码,运行分类训练脚本:
python3 easyai/train_task.py -t classify -i /home/wfw/easy_data/ImageSets/train.txt -v /home/wfw/easy_data/ImageSets/val.txt -m classnet
出现下面报错:
import amp fail!
Traceback (most recent call last):
File "easyai/train_task.py", line 5, in <module>
from easyai.tasks.utility.registry import REGISTERED_TRAIN_TASK
File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/__init__.py", line 2, in <module>
from . import det2d
File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/det2d/__init__.py", line 3, in <module>
from . import detect2d_train
File "/home/wfw/workspace/EDGE/easy_ai/easyai/tasks/det2d/detect2d_train.py", line 6, in <module>
from easyai.data_loader.det2d.det2d_train_dataloader import get_detect2d_train_dataloader
File "/home/wfw/workspace/EDGE/easy_ai/easyai/data_loader/det2d/det2d_train_dataloader.py", line 15, in <module>
from easyai.tools.sample_tool.create_detection_sample import CreateDetectionSample
File "/home/wfw/workspace/EDGE/easy_ai/easyai/tools/sample_tool/create_detection_sample.py", line 13, in <module>
from easyai.tools.sample_tool.sample_info_get import DetectionSampleProcess
ImportError: cannot import name 'DetectionSampleProcess'
@lpj0822
已经更新