PascalVOC label scripts are updated
| | |
| | | |
| | | |
| | | def convert(size, box): |
| | | dw = 1./size[0] |
| | | dh = 1./size[1] |
| | | x = (box[0] + box[1])/2.0 |
| | | y = (box[2] + box[3])/2.0 |
| | | dw = 1./(size[0]) |
| | | dh = 1./(size[1]) |
| | | x = (box[0] + box[1])/2.0 - 1 |
| | | y = (box[2] + box[3])/2.0 - 1 |
| | | w = box[1] - box[0] |
| | | h = box[3] - box[2] |
| | | x = x*dw |
| | |
| | | convert_annotation(year, image_id) |
| | | list_file.close() |
| | | |
| | | os.system("cat 2007_train.txt 2007_val.txt 2012_train.txt 2012_val.txt > train.txt") |
| | | os.system("cat 2007_train.txt 2007_val.txt 2007_test.txt 2012_train.txt 2012_val.txt > train.all.txt") |
| | | |
| | |
| | | |
| | | |
| | | def convert(size, box): |
| | | dw = 1./size[0] |
| | | dh = 1./size[1] |
| | | x = (box[0] + box[1])/2.0 |
| | | y = (box[2] + box[3])/2.0 |
| | | dw = 1./(size[0]) |
| | | dh = 1./(size[1]) |
| | | x = (box[0] + box[1])/2.0 - 1 |
| | | y = (box[2] + box[3])/2.0 - 1 |
| | | w = box[1] - box[0] |
| | | h = box[3] - box[2] |
| | | x = x*dw |
| | |
| | | { |
| | | if (mat.data == NULL) return std::shared_ptr<image_t>(NULL); |
| | | |
| | | cv::Size s = mat.size(); |
| | | if (get_net_width() != s.width || get_net_height() != s.height) { |
| | | cv::Size network_size = cv::Size(get_net_width(), get_net_height()); |
| | | cv::Mat det_mat; |
| | | cv::resize(mat, det_mat, cv::Size(get_net_width(), get_net_height())); |
| | | return mat_to_image(det_mat); |
| | | } |
| | | if (mat.size() != network_size) |
| | | cv::resize(mat, det_mat, network_size); |
| | | else |
| | | det_mat = mat; // only reference is copied |
| | | |
| | | return mat_to_image(mat); |
| | | return mat_to_image(det_mat); |
| | | } |
| | | |
| | | static std::shared_ptr<image_t> mat_to_image(cv::Mat img_src) |