Vào thẳng nội dung

Đo mực nước

Tổng quan

Sử dụng SAM và SOLOv2 để phát hiện và đo mức nước

Vui lòng cài đặt ecos_core-1.1.0 hoặc phiên bản mới nhất và AdelaiDet-0.2.0 trước. 

pip install ecos_core-1.1.0.zip
pip install AdelaiDet-0.2.0.zip

Chuẩn bị bộ dữ liệu của bạn

# cấu trúc bộ dữ liệu của bạn nên như sau
datasets/
    -annotations/
        - instances_train2017.json
        - instances_val2017.json
    -train2017/
        -*.jpg
    -val2017/
        -*.jpg

Cách sử dụng

1. Để huấn luyện, chạy:

python train.py --opt-solo "<ĐƯỜNG_DẪN_TỚI_TẬP_TIN_OPT>"

Ví dụ:

python train.py --opt-solo ./opt.json
  • ĐƯỜNG_DẪN_TỚI_TẬP_TIN_OPT là đường dẫn đến tệp opt.json

2. Để kiểm tra, chạy:

python predict.py --input-path "<PATH_TO_TEST_IMG>" --opt-sam "<PATH_TO_SAM_OPT_FILE>" --opt-solo "<PATH_TO_SOLOV2_OPT_FILE>" --output-path "<PATH_TO_OUTPUT_IMG>"

Hoặc 

python predict.py --input-path ./../../../riwa_v2/test/images/ADE_train_00004363.jpg --opt-sam ./../../ecos_core/SAM/opt.json --opt-solo ./opt.json --output-path test.png
  • ĐƯỜNG_DẪN_TỚI_HÌNH_ẢNH_KIỂM_TRA là nơi có tệp hình ảnh
  • ĐƯỜNG_DẪN_TỚI_HÌNH_ẢNH_KẾT_QUẢ là nơi sẽ lưu kết quả
  • ĐƯỜNG_DẪN_TỚI_TẬP_TIN_OPT_SAM là đường dẫn đến tệp opt.json của SAM
  • ĐƯỜNG_DẪN_TỚI_TẬP_TIN_OPT_SOLOV2 là đường dẫn đến tệp opt.json của SOLOv2
  • Nếu kết quả dự đoán không tốt, hãy xem xét điều chỉnh ngưỡng hoặc thêm dữ liệu